Linux系统下pytorch官网conda命令安装pytorch的版本是cpu

DanielFaster

已于 2022-04-17 10:12:24 修改

阅读量3.6k

点赞数

分类专栏： python pytorch 文章标签： python

于 2022-03-31 17:03:22 首次发布

本文链接：https://blog.csdn.net/binbinczsohu/article/details/123876539

版权

26 篇文章 2 订阅

订阅专栏

12 篇文章 0 订阅

订阅专栏

环境说明

Ubuntu 18.04.5 LTS (GNU/Linux 4.15.0-173-generic x86_64)

显卡驱动

CUDA Version: 10.1

NVIDIA-SMI 418.56

Driver Version: 418.56

在用一个环境的时候出现了The NVIDIA driver on your system is too old，这个问题有两种解决办法，一种是升级显卡驱动，另一种是调整pytorch的版本。

升级显卡驱动是多麻烦的一件事情呀，能不升级，尽量不升级。而且有些软件他就只能用低版本的，所以我采取了第二种办法，调整pytorch的版本。

上了pytorch官网发现cuda10.1已经被人抛弃了问题二：用pytorch官网conda命令安装的pytorch的版本是cpu

在pytorch 官网再看下之前的版本的安装命令，发现有cuda10.1 但是按照命令执行，发现是conda搜索到的版本居然是cpu的

按照官网的命令执行，搜索到的版本居然是cpu的

使用官网的方法，是不行的

# CUDA 10.1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

将官网的命令改成10.1 也是不行

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

换个搜索引擎搜出来是这个链接，操作步骤如下

First see if CUDA 10.1 is installed
cat /usr/local/cuda/version.txt
[For pip] Run pip3 install with specified version and -f. Here we will install 1.7.0. You can also install 1.3.0, 1.3.1, 1.4.0, 1.5.0, 1.5.1, 1.6.0.
pip3 install torch==1.7.0 torchvision==0.8.1 -f https://download.pytorch.org/whl/cu101/torch_stable.html

Use pip if you are using Python 2.

Note: PyTorch currently supports CUDA 10.1 up to the latest version (Search torch- in https://download.pytorch.org/whl/cu101/torch_stable.html).
[For conda] Run conda install with cudatoolkit
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
如果上面方法不行出现这种情况，查询得到的版本仍然是cpu的，可以看第五步换另一条命令

使用新命令(从官网查询）（22/4/17更新）

# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

Verify PyTorch is installed
Run Python with

import torch x = torch.rand(5, 3) print(x)
Verify PyTorch is using CUDA 10.1
import torch torch.cuda.is_available()