Pytorch2.3.1+cu121-GPU速装简单明了
即今江海一归客,他日云霄万里人
第一步
首先检查自己的CUDA版本进入NVIDIA控制面板
接下来我们就可以根据CUDA版本选择torch版本
参考网站
CUDA和torch对应关系看这个: https://dev-discuss.pytorch.org/t/pytorch-release-2-4-1-final-rc-is-available/2422
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
然后命令行显示如下:
第二步
pip install torch==2.3.1+cu121 torchvision==0.18.1+cu121 torchaudio==2.3.1+cu121 -f https://download.pytorch.org/whl/torch_stable.html
CMD命令行结果如下: