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 版本错误，请尝试 installing an earlier version PyTorch 的。

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

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

关于python - 如何安装启用 CUDA 的 Pytorch 1.3.1，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/59563220/