问题描述:
利用清华源镜像网站下载cuda版本的pytorch时,按照网上的教程,不管怎么输入指令,最终都会报错:
1.输入如下指令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
可能由于是境外网络的原因,出现了如下报错:
Collecting package metadata (repodata.json): failed
2.按照网上的教程,去掉 -c及其后面的内容:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8
同样出现报错,但这次报错与上面的报错不太一样:
The following packages are incompatible
└─ pytorch-cuda 11.8** is not installable because there are no viable options
├─ pytorch-cuda 11.8 would require │
└─ cuda-cudart >=11.8,<12.0 ,
3.我后面又试了试只删去-c pytorch,保留-c nvidia,同样无法下载。
4.我怀疑是由于清华源网站并没有跟随pytorch官网使用最新的下载指令,而是仍在使用原指令下载。于是我又尝试使用先前旧的指令下载,如下所示
conda install pytorch torchvision torchaudio cudatoolkit=11.8
结果:仍然报错,无法下载。
问题分析:
利用清华源镜像网站下载pytorch库时,看了网上的博客,大多都是2022年之前的,而最新版本的pytorch的下载指令却有所变化,因此可能清华源镜像网站可能并不能对比较新版本cuda的pytorch下载。
问题解决:
通过如下网站直接手动下载,找到自己需要的pytorch版本(记住cuxxx开头的表示cuda版本,cpxxx开头的为你的python版本,一定要对应好自己的python和cuda版本!!!否则无法下载),下载后放到自己设定的路径下。
下载网站连接如下:
https://download.pytorch.org/whl/torch_stable.html
比如我的cuda版本是11.8,python版本是3.11,系统是64位win,我要下载的两个文件如下:
(一个torch 版本2.1.1,一个torchvision 版本0.16.1)
(【注意】:要对应的版本如下
①.python的版本要对应
②.cuda的版本要对应
③.系统的版本要对应
④.torch和torchvision的版本也要对应
(例如我下载的cuda版本是2.1.1,那么torchvision版本就必须是0.16.1,你可以根据我的版本倒推出你的cuda版本所对应的torchvision版本) ):
(但凡有一个版本没对应,就无法下载,会报错,亲测!!)
我下载到了C:\pytorch\文件夹下(把两个文件统一下载到了一个文件夹下)。
下载好后,在终端(terminal)窗口用pip install安装。
首先先cd到你所下载的文件夹目录下:
cd C:\pytorch
然后用pip install文件名进行安装:
1.安装torch
pip install "torch-2.1.1+cu118-cp311-cp311-win_amd64.whl"
2.安装torchvision
同理:
pip install "torchvision-0.16.1+cu118-cp311-cp311-win_amd64.whl"
测试:
输入以下代码:
>>> import torch
>>> torch.cuda.is_available()
结果为true,证明torch安装成功。
输入代码:
>>> import torchvision
>>> torchvision.__version__
显示torchvision的版本,证明torchvision安装成功
参考网站:
Pycharm搭建CUDA,Pytorch教程(匹配版本,安装,搭建全保姆教程)_cuda12.3对应的pytorch版本-CSDN博客