1. 查看CUDA版本:
nvidia-smi
由图可知我们的CUDA版本为11.6
2. 打开这个网站,直接查看并选择适配你CUDA版本的Pytorch配置(非常直观明了)
Previous PyTorch Versions | PyTorch
举个CUDA11.6的例子:
# CUDA 11.6
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia
总之这一步就直接告诉你了你要安装的pytorch、torchvision的版本到底是什么。如果一下子装不上的话就一个个装,耐心点,遇到问题或者装的很慢的话去优快云上看看其他人时怎么装的。反正保证你的版本号正确就可以了
3. 全部安装完成后再次检查版本号即可