【2024最新】下载cuda版本的pytorch库时清华源不可用的问题(已解决)

文章讲述了用户在使用清华源镜像下载PyTorch时遇到的问题,因为教程过时导致下载失败。作者发现需要根据最新PyTorch官网的指示下载对应CUDA版本,并强调版本对应的重要性。最后,作者提供了手动下载并正确安装的方法,确保了PyTorch和torchvision的成功安装。
摘要由CSDN通过智能技术生成

问题描述:

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

  • 30
    点赞
  • 37
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值