Почему Tensorflow 2.1 не распознает мой графический процессор? - PullRequest
0 голосов
/ 06 мая 2020

Я пытаюсь установить Tensorflow 2 на машине windows, которая, как я проверял, имеет графический процессор с поддержкой Cuda (Nvidia Quadro RTX 3000). Я установил необходимые библиотеки CUDA и SDK и добавил в свой путь (см. Ниже): enter image description here.

Я установил tenorflow и tensorflow-gpu через pip. Я проверил свою настройку следующим образом:

gpus = tf.config.list_physical_devices('GPU')
result = tf.config.list_physical_devices()
print(tf.__version__)
print(result)
print('Built with cuda: ', tf.test.is_built_with_cuda())
print('Built with gpu support: ', tf.test.is_built_with_gpu_support())
print('is gpu available: ', tf.test.is_gpu_available())
print('gpus: ', gpus)

Тем не менее, я получаю следующие результаты:

2.1.0
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
Built with cuda:  True
Built with gpu support:  True
WARNING:tensorflow:From <ipython-input-1-601dc453d590>:14: is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.config.list_physical_devices('GPU')` instead.
is gpu available:  False
gpus:  []

Другие вопросы по этой проблеме касаются проблем с настройкой тензорного потока, который не кажется быть здесь. Почему тензорный поток все еще не видит мой графический процессор? Я что-то пропустил?

1 Ответ

1 голос
/ 07 мая 2020

TensorFlow 2.1 пока не поддерживает CUDA 10.2.

Вы должны использовать CUDA 10.1, как написано в docs .

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...