如何在启用CUDA的情况下安装Pytorch 1.3.1
我在我的Ubuntu 16.04系统上有一个conda环境。
使用以下命令安装Pytorch时：
conda install pytorch然后我尝试运行我需要的脚本，我得到了错误消息：
raise AssertionError("Torch not compiled with CUDA enabled")通过查看论坛，我发现这是因为我安装的Pytorch没有CUDA支持。
然后我试着：
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版本，以便与我的系统兼容？
回答 3
Stack Overflow用户
发布于 2020-01-12 04:12:08
鉴于您的系统运行的是Ubuntu 16.04，它附带安装了glibc。您可以通过键入ldd --version来检查您的版本。
请记住，CentOS which runs glibc version 2.17.上的PyTorch is compiled
然后检查系统nvcc --version上安装的CUDA 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
Stack Overflow用户
发布于 2020-01-12 04:15:51
对于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 pytorchStack Overflow用户
发布于 2020-07-19 18:40:47
我不确定你是否已经解决了你的问题，但我之前遇到了完全相同的问题，因为我试图在集群上安装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，您可能希望将其设置为您有权访问的其他目录)。希望它能为你工作
https://stackoverflow.com/questions/59563220
复制相似问题
腾讯云开发者
