当前位置:   article > 正文

cuda cudnn pytorch 的下载方法(anaconda)

cuda cudnn pytorch 的下载方法(anaconda)

前言

记录一下近日下载pytorch时遇到的坑

cuda

查看当前可支持的最高cuda版本

快捷键
win+R
cmd 
NVIDIA-smi
  • 1
  • 2
  • 3
  • 4

cmd命令行内输入NVIDIA-smi即可查看当前NVIDIA显卡的driver version(显卡驱动),和可安装的最高cuda版本(最好别太高,也别太低)

在这里插入图片描述

显卡驱动更新

显卡驱动太旧的话,要下载好新版本的驱动

笔者用的是以下软件更新的

在这里插入图片描述

下载cuda

下载前还要去看一下目前pytorch适配的最高cuda版本(很重要!)

https://pytorch.org/get-started/locally/
在这里插入图片描述

以pytorch网页上的cuda版本为主!
我就掉进了这个坑里
图里显示的cuda版本最高为12.1,但我之前却下了12.4版本的,最后搞了好久

cuda官网

cudnn

cudnn官网

选择与自己的cuda版本对应的cudnn版本即可

pytorch

上面第一个网址

配置虚拟环境

打开anacona自带的
在这里插入图片描述

创建虚拟环境

个人推荐python3.10版本

conda create -n pytorch(环境名) python=3.10.14(版本名)

在这里插入图片描述

出现以下图图,即代表母子平安
在这里插入图片描述

激活虚拟环境

conda activate 环境名
在这里插入图片描述

1.直接下载

根据自己的情况选好,即可复制

Run this Command

上的命令即可
如果找不到对应cuda版本的话,可以点击

install previous versions of PyTorch

这个链接去寻找属于自己的神奇宝贝!

2.conda 下载(清华源,下载速度慢的看过来)

由于软件过大,使用conda安装软件时可能会遇到

下载中断
龟速下载

而且conda没有断点下载这一说法,下载已中断的话,就得重新来过了

添加清华镜像channel

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
  • 1
  • 2
  • 3
  • 4

在这里插入图片描述

我之前已经添加过了,所以再次添加会报错(无伤大雅,区区致命伤)

下载

conda install pytorch torchvision torchaudio pytorch-cuda=12.1
在这里插入图片描述

等亿会,出现

done

后即可开始检验是否成功下载了

python
import torch
torch.cuda.is_available()

下载失败

如果最终结果为true,则大功告成啦
如果不是,那只能再试亿下下啦

如果出现以下图像,恭喜你,孩子很健康…
在这里插入图片描述

下载失败啦
我就是掉进了这个坑(cuda12.4版本过于先进,有人说可以向下兼容cuda12.1版的pytorch,但我这不行)

下载失败解决办法

1.浑水摸鱼,风浪越大鱼越贵!

搞了老半天,还是失败了,气得我当时就气了一个下午!
结果晚上我捣鼓大模型的时候,发现下载绘世模拟器的时候,阴差阳错的下好了,emm…
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

2.江湖凶险,大侠请重新来过

不知道可不可行,个人想法
找个办法卸载掉cuda(应该挺难搞的)和cudnn,重新再来一次捏
因为当时已经通过绘世模拟器下好了,就懒得亲手实践了(主要是这边网速不好啊!!!)

windows cuda更新教程

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/360086
推荐阅读
相关标签
  

闽ICP备14008679号