PyTorch 安装

本文基于conda安装，请确保已经安装好Anaconda，可参考上一篇文章安装Anaconda。

一、确定本机CUDA版本

基于N卡，基于N卡，基于N卡

nvidia-smi  #N卡用此命名查看，N卡如果没有此命令，先去更新显卡驱动

二、安装

前往官网安装适合自己的版本。确保你当前的版本大于等于你选择的版本即可，向下兼容。
官网地址：https://pytorch.org/get-started/locally/

三、复制命令进conda执行

先激活CONDA的环境，然后输入

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

记得确认输个Y，进入以下界面后就可以慢慢等了。