如何在启用 CUDA 的情况下安装 Pytorch 1.3.1答案

【问题标题】：How do I install Pytorch 1.3.1 with CUDA enabled如何在启用 CUDA 的情况下安装 Pytorch 1.3.1
【发布时间】：2020-04-21 02:40:30
【问题描述】：

我的 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 以便它适用于我的系统？

【问题讨论】：

你的 cuda 版本是多少？你试过9.0 或10.0 吗？你的操作系统是什么？
@SzymonMaszke 您是否需要在您的操作系统上安装 CUDA 才能与 conda + Pytorch 一起使用？昨天我听说 Pytorch conda 安装带有它自己的 CUDA。
你试过conda install pytorch torchvision cudatoolkit=9.2 -c pytorch吗？
@user1551817 你能发布apt policy libc6的输出吗？
@user1551817 试试apt-cache search libc 看看有哪些版本可用。我建议从那里安装，但如果它不可用，我想你可以尝试从这个频道安装：conda install -c pwwang glibc214 。

标签： python pytorch conda

【解决方案1】：

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

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

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

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

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

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

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

【讨论】：

这会安装最新的 pytorch，而不是专门安装 1.3

【解决方案2】：

对于 CUDA 10.1：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

对于 CUDA 9.2：

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

对于没有 CUDA：

conda install pytorch torchvision cpuonly -c pytorch

【讨论】：

【解决方案3】：

不确定您是否解决了问题，但我之前也遇到过同样的问题，因为我试图在集群上安装 pytorch，但我没有 root 访问权限。您需要将glibc下载到您的目录并将环境变量LD_LIBRARY_PATH设置为您本地的glibc https://stackoverflow.com/a/48650638/5662642。

要在本地安装 glibc，我会将您指向我阅读的这个线程以解决我的问题

https://stackoverflow.com/a/38317265/5662642（而不是在安装时设置 --prefix=/opt/glibc-2.14，您可能希望将其设置为您有权访问的其他目录）。希望对你有用

【讨论】：