一个小时前,我安装了最新的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驱动程序,但仍然无法正常工作,所以我重新开始。