安装操作总览
在安装pytorch之前,请先查看当前电脑中已安装的的CUDA版本(在终端输入 nvcc -V
查看)
-
进入官网Previous PyTorch Versions | PyTorch,查找到对应的系统与版本
-
选择安装方式:pip or conda.(笔者推荐在自己建立的conda虚拟环境中进行安装)
-
创建好conda虚拟环境后,复制官网中的命令即可下载
-
注意,使用conda下载的时候不要加额外的channels,只保留default即可
-
总体安装用时大概五分钟
-
安装实例
-
首先,我们查看CUDA版本
可以看出我的CUDA版本为11.7
-
其次,进入Pytorch官网:Previous PyTorch Versions | PyTorch,查找CUDA与Pytorch的版本对应关系。方法:在Pytorch官网中Ctrl+F查找
11.7
(即CUDA版本号)
可以看到CUDA-11.7支持许多版本的Pytorch
-
最后,我想安装的Pytorch版本是1.13.1
-
创建conda虚拟环境
这里需要注意,在创建虚拟环境之前,需要查看Python版本和Pytorch版本的对应关系,参看官方文档:pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)可以了解Python版本和Pytorch版本的对应关系。
我们进入上述网站查看版本对应关系:
-
根据版本对应关系,我们发现torch1.13对应的Python版本需大于等于3.7.2,同时小于等于3.10。故我们执行以下命令,以创建一个名为<env_name>
的同时具有Python3.8的虚拟环境
conda create -n <env_name> python=3.8
-
安装Pytorch:
在终端输入命令
conda activate <env_name> conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia # 第二行命令是复制Pytorch官网中的命令,不同的Pytorch版本和不同的CUDA版本会有不同的指令!!!
即可进行安装
-
在安装过程中,尤其需要注意的是,在下载的时候conda的下载频道只保留default频道。
以下是对conda的下载频道的一些操作:
-
查看conda的下载频道:
conda config --show channels
-
假设频道名为
<channel_name>
,则删除此频道的指令为conda config --remove channels <channel_name>
. -
将conda的下载频道直接恢复为默认(即只保留default)的指令为
conda config --remove-key channels
.
-
-