☰

如何安装启用了 CUDA 的 Pytorch 1.3.1

tag-icon • tag-icon python cuda
如何安装启用了 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适合我的系统？

答案1

使用官方 anaconda 安装程序时，你只能从有限的预构建 pytorch 版本中进行选择https://pytorch.org/get-started/locally/（然后当然选择那里的 cuda 选项）。

如果您想要一个不再提供的特定版本，则需要从源代码安装它。请参阅以下示例了解如何执行此操作（虽然是针对 Windows 的情况，但只是开始）如何在 Windows 10 上使用 anaconda prompt 从源代码安装 pytorch（为旧 gpu 的已弃用的 CUDA cc 3.5 启用了 cuda）？。

相关内容