全新安装的ubuntu上的Pytorch找不到我的GPU

时间:2018-12-08 16:05:35

标签: python-3.x anaconda gpu nvidia pytorch

一个小时前,我安装了最新的ubuntu 18。我可能在20分钟前安装了最新的anaconda 3.7。然后我继续使用

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上运行,我的最后一次尝试是几天前使用pip install在Lubuntu上,我安装了一些Nvidia驱动程序,但仍然无法正常工作,所以我重新开始。

0 个答案:

没有答案