Pytorch / CUDA - не удается найти графический процессор - PullRequest
Новая
       32

Pytorch / CUDA - не удается найти графический процессор

0 голосов
/ 09 ноября 2018

Я пытаюсь использовать мой графический процессор для обучения сети. Я установил PyTorch и CUDA с анакондой. У меня на компьютере NVIDA GTX 860M, который я хотел бы использовать.

При использовании команды: 

$ lspci | grep NVIDIA

Я получаю вывод: 

01:00.0 3D controller: NVIDIA Corporation GM107M [GeForce GTX 860M] (rev a2)

Итак, графическая карта есть. Но в записной книжке Jupyter есть следующие строки: 

print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.version.cuda)

Возврат: 

0
False
9.0.176

Я попытался обновить последнюю версию драйвера Cuda с помощью conda, но это не помогло. Моя версия PyTorch - 0.4.1.post2. Я не мог найти ответ, просматривая различные онлайн-форумы. Я на Ubuntu 18.04.1 LTS. Заранее спасибо!

...