python - Ubuntu 17.10(GPU)上安装Pytorch的问题

Question

我想在我的计算机上使用 PyTorch 及其 GPU 计算。

我有一台运行 Ubuntu 17.10 的计算机。计算机 (Alienware m17x) 有两个显卡:

• 集成的 Intel Ivybridge Mobile
• Nvidia Geforce 675M。

为了安装 PyTorch，我按照 PyTorch 网站 pytorch.org 上的说明进行操作

1) 我使用 deb 文件安装了 CUDA 9:https://developer.nvidia.com/cuda-downloads

=> Linux/x86_64/Ubuntu/17.04/deb (local)

2)我使用 conda 命令行安装了 Pytorch: conda install pytorch torchvision cuda90 -c pytorch

这两个步骤都没有给我返回任何类型的错误。

我重新启动了计算机。显然这两张卡已被检测到:

$ lspci | grep -i vga
00:02.0 VGA compatible controller: Intel Corporation 3rd Gen Core    processor Graphics Controller (rev 09)
01:00.0 VGA compatible controller: NVIDIA Corporation GF114M [GeForce GTX 675M] (rev a1)

但显然驱动程序或 CUDA 本身有问题。 nvidia-Detector 不会返回任何内容:

$ nvidia-detector 
none

并且pytorch不能使用cuda:

[1]: import torch
In [2]: torch.cuda.is_available()
Out[2]: False

你能帮我吗？如有必要，我可以提供其他信息，但我不确定哪些信息可能相关。

Answer 1

如果您按如下方式安装 pytorch，则无需安装 cuda 即可将 GPU 与 pytorch 一起使用:pytorch 二进制文件包含所有必需的 cuda 库。

因此，在安装 pytorch 时选择哪个 cuda 版本风格也并不重要。通常人们可能会想要最新版本，但在需要使用旧 GPU 的情况下，旧 cuda 版本附带的 pytorch 二进制文件可能是唯一仍然支持该 GPU 的版本。

如果没有检测到 GPU，那么这可能与 CUDA 库无关，而是与您的内核驱动程序有关。确保您的系统安装了经过测试的最新 NVIDIA 专有内核驱动程序。

可能有点令人困惑的是，可以在任何系统上安装支持 cuda 的 pytorch 二进制文件，包括没有 GPU 或有 GPU 但没有安装系统驱动程序的系统。在您尝试实际使用 GPU 并调用 .cuda()

之前，这一切正常。