python - 如何在启用CUDA的情况下安装Pytorch 1.3.1

我的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，以便它与我的系统兼容？

最佳答案

假设您的系统运行的是Ubuntu 16.04，则安装了glibc。您可以通过键入ldd --version来检查您的版本。

请记住，PyTorch is compiled上的CentOS which runs glibc version 2.17.

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

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

如果出现glibc版本错误，请尝试PyTorch的installing an earlier version。

如果以上两个选项都不起作用，请尝试installing PyTorch from sources.