时间戳

首页 > 解决方案 > 如何在启用 CUDA 的情况下安装 Pytorch 1.3.1

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

标签： pythonpytorchconda

解决方案

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

请记住，PyTorch 是在运行 glibc 2.17 版的 CentOS 上编译的。

然后检查系统上安装的CUDA 版本nvcc --version

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

如果您收到 glibc 版本错误，请尝试安装较早版本的 PyTorch。

如果上述选项都不起作用，请尝试从源代码安装 PyTorch。

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

推荐阅读