最近要安装gpu版本的pytorch,使用conda和pip安装各种报错,国外的镜像下载报各种乱七八糟的错,国内的有的包找不到,有的下载也报错。不得不说pip和conda的下载是真玄学,最终还是通过浏览器下载包安装搞定。
pytorch官网:https://download.pytorch.org/whl/torch_stable.html
直接进官网的下载界面,选择相应的torch和torchvision版本,注意版本对应,我这里选择的是
torch-2.0.1+cu117-cp39-cp39-win_amd64.whl和torchvision-0.15.2+cu117-cp39-cp39-win_amd64.whl,python版本为3.9.0。
下载完成后在prompt中进入到下载的文件夹,直接
pip install torch-2.0.1+cu117-cp39-cp39-win_amd64.whl
pip install torchvision-0.15.0+cu117-cp39-cp39-win_amd64.whl
由于本地中存在不会从镜像中下载而是直接解压,等待安装完成即可。