如何安装启用 CUDA 的 Pytorch 1.3.1

Question

我的 Ubuntu 16.04 系统上有一个 conda 环境。

当我使用以下命令安装 Pytorch 时：

conda install pytorch

我尝试运行我需要的脚本，但收到错误消息：

raise AssertionError("Torch not compiled with CUDA enabled")

通过查看论坛，我发现这是因为我安装了没有 CUDA 支持的 Pytorch。

然后我尝试：

conda install -c pytorch torchvision cudatoolkit=10.1 pytorch

但现在我收到错误：

    from torch.utils.cpp_extension import BuildExtension, CUDAExtension
  File "/home/username/miniconda3/envs/super_resolution/lib/python3.6/site-packages/torch/__init__.py", line 81, in <module>
    from torch._C import *
ImportError: /lib64/libc.so.6: version `GLIBC_2.14' not found

所以看来这两个安装正在安装不同版本的 Pytorch（？）。第一个似乎有效的版本是 Pytorch 1.3.1。

我的问题：如何在启用 CUDA 的情况下安装 Pytorch，但确保其版本1.3.1适合我的系统？

Answer 1

鉴于您的系统运行的是 Ubuntu 16.04，它安装了 glibc。您可以通过输入来检查您的版本ldd --version。

请记住，PyTorch 是在运行 glibc 版本 2.17 的 CentOS上编译的。

然后检查系统上安装的CUDA版本nvcc --version

然后按如下方式安装 PyTorch，例如，如果您的 cuda 版本是 9.2： conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

如果出现 glibc 版本错误，请尝试安装早期版本的 PyTorch。

如果以上选项均不起作用，请尝试从源安装 PyTorch。

如果您想设置要安装的特定 PyTorch 版本，请在以下命令中将其设置为 <version_nr>： conda install pytorch=<version_nr> torchvision cudatoolkit=9.2 -c pytorch