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



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

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

  • 集成的英特尔常春藤桥移动式
  • 英伟达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安装pytorch火炬视觉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 
none

而 pytorch 不能使用 cuda:

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

你能帮我吗?如有必要,我可以提供其他信息,但我不确定什么是相关的。

如果您像这样安装 pytorch,则无需安装 cuda 即可将 GPU 与 pytorch 一起使用: pytorch 二进制文件包含所有必要的 cuda 库。

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

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

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

1.如果你有两个 GPU,不要使用 CUDA deb 文件。请改用运行文件,因为在询问是否安装 openGL 和 x-server 选项时,您需要选择 N。

2.更重要的是,Cuda9.0不支持17.10。Cuda 9.0 仅支持 unbuntu 17.04 和 16.04

3.只有 cuda9.2 支持 ubuntu17.10 https://developer.nvidia.com/cuda-92-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1710

4.请务必仔细阅读 Linux 安装指南。我在这里写了一个相关的答案。 适用于 Ubuntu 18.04 的 Nvidia Tesla K40M GPU 驱动程序设置,用于 Matlab

最新更新