问题描述:

利用清华源镜像网站下载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博客

Logo

欢迎加入西安开发者社区!我们致力于为西安地区的开发者提供学习、合作和成长的机会。参与我们的活动,与专家分享最新技术趋势,解决挑战,探索创新。加入我们,共同打造技术社区!

更多推荐