☰

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

tag-icon • tag-icon nvidia python anaconda
新安装的 ubuntu 上的 Pytorch 找不到我的 GPU

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

答案1

刚刚找到一个解决方案，也许......检查https://ubuntuforums.org/showthread.php?t=2377560了解详情。

也许您使用的是混合显卡，这可能会导致驱动程序安装错误。还请检查 BIOS 中的安全启动是否已禁用。

在终端中运行：

sudo apt purge nvidia*
sudo apt update
sudo ubuntu-drivers autoinstall
sudo apt full-upgrade

重新启动并再次检查。

答案2

如我所见，您已经完成了 PyTorch 的“正常”安装，目前是 CUDA 9。

我刚刚检查了NVIDIA CUDA 9 文档，Ubuntu 18 在这种情况下不受支持。如果你想在 Ubuntu 18 上使用 CUDA，那么你必须使用CUDA 10根据文献。

答案3

首先确保 CUDA 已正确安装。您可以使用https://developer.nvidia.com/cuda-zone安装系统的最新版本

相关内容