Я установил cuda, cudann и tensorflow-gpu в среде jupyter, и после этого я пытаюсь проверить, есть ли у меня поддержка gpu в этой среде, но в list_local_devices он не показывает мне gpu. У меня в ноуте стоит geforce 1050 ti gpu.
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
import tensorflow as tf
tf.test.is_gpu_available()
Я получаю такой ответ: -
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5705862024723076222
]
False






возможно, лучше установить путь CUDA, например:
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
и добавьте его на свой локальный ~/.bashrc.
Также убедитесь, что CUDA установлен правильно, как описано здесь.
Команда
nvidia-smi
Должны быть показаны аппаратное обеспечение графического процессора, версия CUDA и драйвер NVIDIA-SMI.
Надеюсь, это поможет!