本文基于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,进入以下界面后就可以慢慢等了。