Python, Pytorch, возникли проблемы при использовании cuda

Question

ОС: Windows 10

Python 3.7.4 (Conda)

Графический процессор: GTX 980

Итак, я установил инструментарий CUDA v.10.1, получил соответствующие файлы cuDNN, установили версию Pytorch с поддержкой cuda 10.1 и в дополнение к этому я обновил мои драйверы GPU. но если я сейчас попробую проверить, доступен ли мой графический процессор с torch.cuda.is_available(), он все равно получит False. Любые идеи ??

Answer 1

Необходимо убедиться, что python работает на графическом процессоре, а не на процессоре.

Это можно сделать, запустив IDE на графическом процессоре, выбрав один из вариантов, щелкнув правой кнопкой мыши. Просто щелкните правой кнопкой мыши на вашей IDE и выберите Run with graphics processer. Как показано на рисунке ниже.

Кроме того, вы можете go до Nvidia Control Panel > Manage 3D Settings > Program Settings и настроить графический процессор по умолчанию Python и ваш IDE для NVIDIA GPU. Панель управления можно увидеть ниже.