Pytorch в новой версии Ubuntu не может найти мой GPU - PullRequest
0 голосов
/ 08 декабря 2018

У меня установлена ​​последняя версия Ubuntu 18 всего час назад.У меня установлена ​​последняя версия anaconda 3.7, может быть, 20 минут назад.Затем я продолжил использовать

conda install pytorch torchvision -c pytorch

, но когда я открываю использовать python3, он все равно дает мне

alex@alex:~$ python3
Python 3.7.0 (default, Jun 28 2018, 13:15:42) 
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False
>>> 

Конечно, у меня в компьютере есть GPU:

alex@alex:~$ lspci | grep VGA
01:00.0 VGA compatible controller: NVIDIA Corporation GP107 [GeForce GTX 1050 Ti] (rev a1)

Я также перезагрузил компьютер после установки pytorch, и он ничего не изменил.Это моя вторая попытка заставить GPU работать на pytorch, моя последняя попытка была на Lubuntu несколько дней назад с использованием установки pip, и я установил некоторые драйверы Nvidia, но все равно не смог заставить его работать, поэтому я начал заново.

...