conda环境下安装Pytorch

安装操作总览

在安装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.