pytorch的离线安装

pytorch的离线安装

第一步，打开anaconda命令行，输入nvcc -version 查看cuda版本，进入python查看python 版本

这里我的是cuda11.1 版本

python版本为3.8.8

第二步，打开离线网站，找到对应的pytorch版本和torchvision版本

网站链接：https://download.pytorch.org/whl/torch_stable.html
选择对应版本进行下载，以我的为例

在anaconda powershell中输入对应命令

安装完成！

一定要找到对应的torch和torchvision版本，否则跑代码会出现warning或者无法使用gpu的情况。