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

来自torch.utils.cpp_扩展导入BuildExtension，CUDAExtension
文件“/home/username/miniconda3/envs/super_resolution/lib/python3.6/site packages/torch/__init___;.py”，第81行，in
从火炬进口*
ImportError:/lib64/libc.so.6:未找到版本'GLIBC_2.14'

因此，这两次安装似乎安装了不同版本的Pytorch（？）。第一个似乎有效的是Pytorch 1.3.1

---

我的问题：如何在启用CUDA的情况下安装Pytorch，但要确保它的版本是

1.3.1

，这样它才能与我的系统一起工作？

鉴于您的系统运行的是Ubuntu 16.04，它附带了glibc安装。您可以通过键入

ldd--version

检查您的版本

记住这一点

然后检查系统上安装的版本

nvcc--version

然后按照以下步骤安装PyTorch，例如，如果您的cuda版本为9.2：

conda安装pytorch torchvision cudatoolkit=9.2-c pytorch

如果出现glibc版本错误，请尝试PyTorch

如果以上两个选项都不起作用，请尝试

如果要设置要安装的特定PyTorch版本，请按以下命令进行设置： 康达安装pytorch=torchvision cudatoolkit=9.2-c pytorch适用于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

---

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

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

---

（安装时，您可能希望将其设置为您有权访问的其他目录，而不是设置--prefix=/opt/glibc-2.14）。希望它适合您

您的cuda版本是什么？您是否尝试过

9.0

或

10.0

？您的操作系统是什么？@SzymonMaszke您需要在操作系统上安装CUDA才能将其与

conda+Pytorch

一起使用吗？昨天我听说Pytork conda安装有自己的CUDA。您是否尝试过使用

conda安装Pytork torchvision cudatoolkit=9.2-c Pytorch

？@user1551817您能否发布

apt policy libc6

的输出？@user1551817尝试

apt缓存搜索libc

，看看有哪些版本可用。我建议从那里安装，但如果它不可用，我想您可以尝试从以下渠道安装：

conda install-c pwwang glibc214

。这将安装最新的pytorch，而不是1.3