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,即安装完毕!!!