Pytorch最新安装教程（一步到位）

1. 安装anaconda

直接官网安装最新版即可。然后一路next，下图两个选项都点上。
[外链图片转存失败,源站可能有防盗img-niRp src=“C:\Users\92475\AppData\Roaming\Ty-ora\typora-u1erimages\image-202303148262003084.png” referrerpolicy=“no-referrer” alt=“image-C02303:82\4003084”>
ers\92475\AppData\Roaming\Typora\typora-user-images\image-20230318214003084.png)]
接着创建一个pytorch环境，py3.7.2是环境名，python=3.7.2是python版本

创建环境 ： conda create -n py3.7.2 python=3.7.2

激活环境 ： activate py3.7.2

2. 切换清华源

2.1 删除之前的镜像源，恢复默认状态

conda config --remove-key channels

2.2 添加清华镜像源

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

补充：conda的常用命令

查看所有虚拟环境 conda info --env 在显示的环境中，有*的表示当前所在的环境
创建环境 conda create -n envname python=3.6(指定python版本为3.6)
启动环境 conda activate envnane
关闭环境 conda deactivate envname
删除环境 conda remove -n envname --all
克隆环境 本地已有一个环境AAA，若需要创建一个同样的BBB conda create -n BBB --clone AAA

3. 安装mamba

mamba是conda的加速神器，可以用来并行下载和安装，大大加快速度，减少失败几率

conda install mamba -c conda-forge
conda install mamba -n base -c conda-forge

4. 在线安装Pytorch

先查看电脑支持的最大cuda版本，在命令提示符中输入nvidia-smi

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ClmA0Ur0-1679147114667)(C:\Users\92475\AppData\Roaming\Typora\typora-user-images\image-20230318212154210.png)]

从官网找到下载指令，cuda版本不能大于电脑支持的最大版本，同时将conda改为mamba

mamba install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

注意：torch、torchvision与python版本也有关系，所以之前建立的python环境版本也要相符[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-eaYs4Ktt-1679147114667)(C:\Users\92475\AppData\Roaming\Typora\typora-user-images\image-20230318212325330.png)]

5. 验证是否安装成功

进入python环境中：

python

查看torch版本

import torch
print(torch.__version__)

检查Cuda：

torch.cuda.is_available()

显示true,成功结束！

6. 设置python解释器

如果需要使用本虚拟环境在Pycharm进行项目开发，将设置里的Project Interpreter改为相应Anaconda文件目录下的Pytorch虚拟环境中的python.exe文件即可。(如：D:\Anaconda3\envs\pytorch\python.exe）