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的情况。