安装PyTorch详细过程（个人过程仅供参考）

1.安装anaconda

2.创建一个虚拟环境

以上步骤默认已经完毕，毕竟只是记录pytorch的安装过程

3.查看个人电脑CUDA版本

win+r 输入cmd 回车

输入指令 

nvidia-smi

 右上角为该电脑所支持的最高CUDA版本

输入命令

nvcc -V

 图中即为该电脑所安装的CUDA版本

记住该版本

4.进入pytorch官网 PyTorch

点击 get started

选择自己电脑对应的版本，如windows 使用Conda安装 CUDA版本为12.1

之后复制下面图片方框里的命令，即可安装

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

安装包比较大，请耐心等待。

安装完毕后依次输入

python

import torch

torch.cuda.is_available()

返回TRUE，即安装完毕！！！