Блог

У меня Nvidia RTX 2070, и я хочу использовать программирование на графическом процессоре. Я установил Cuda простым способом через Conda, но каким-то образом tensorflow не обнаруживает ни одного устройства с графическим процессором. Моя версия tensorflow 2.3.0.

Я уже пытался использовать следующую команду в приглашении anaconda, но безуспешно.

conda create --name tfgpu python=3.8.5 tensorflow-gpu

Я также попробовал более длинную версию загрузки CUDA toolkit и файла cudnn и копирования, вставки файлов. но, похоже, ничего не работает. Когда я запускаю следующую команду, я всегда получаю 0.

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

Действительно нужна помощь.

Вы можете ознакомиться с этим руководством по установке TensorFlow и тестированию, также обязательно загрузите и установите последние версии библиотек cudnn и Cuda с официального сайта Nvidia. https://towardsdatascience.com/guide-to-conda-for-tensorflow-and-pytorch-db69585e32b8