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

【问题标题】：Problems of Pytorch installation on Ubuntu 17.10 (GPU)在 Ubuntu 17.10 (GPU) 上安装 Pytorch 的问题
【发布时间】：2018-06-25 03:02:44
【问题描述】：

我想在我的计算机上使用 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

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

【问题讨论】：

Ubuntu 17.10 不支持任何当前可用的 CUDA 版本。您是否执行了 CUDA linux 安装指南中的验证步骤？
您的 Nvidia 驱动程序版本是多少？如果 Ubuntu 使用可能导致问题的默认 nouveau 驱动程序。（nvidia-smi 显示驱动版本）

标签： python ubuntu pytorch

【解决方案1】：

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 driver setup for nvidia tesla k40m gpu for use with matlab

【讨论】：

【解决方案2】：

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

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

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

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

【讨论】：