Windows配置深度学习环境:安装Pytorch(自动安装cudn和cudnn+图文+快速+很简单)几分钟搞定!

本文介绍了在Windows环境下,如何检查CUDA版本,使用Anaconda创建深度学习环境,并通过离线方式安装PyTorch,以加速GPU上的深度学习模型训练。作者还分享了如何验证安装是否成功以及安装其他依赖包的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


        因为我要在树莓派上面部署yolov5(已经部署完了),完成一个项目,但是在自己的电脑上训练相关数据集,得到相关的权重文件时,发现它是用CPU跑的,时间很感人,一轮20min左右(电脑是联想拯救者),因此我在这写下部署GPU深度学习环境的一篇博客,希望对你们部署有所帮助。

前提条件: 首先你得有一台有显卡的电脑。

一.查看自己电脑支持的cuda最高版本

#第一步:按下win+R ,输入cmd打开终端
#第二步:输入nvidia-smi
nvidia-smi
#结果如下图,找到自己电脑支持的最高cuda版本

在这里插入图片描述

二.打开Anaconda,进入自己创建的深度学习环境

        没有创建的可以看这一步,之前创建过的可以跳过

#第一步:打开Anaconda Prompt
#第二步:输入下述代码,-n 后面的可以自己起名字,我起的是deeplearn
conda create -n deeplearn python=3.9
#第三步:按要求输入 y 并回车,结果如下图

在这里插入图片描述

三.这用的是离线安装Pytorch,官网安装太慢

1.打开北京外国语大学镜像站

2.找到自己所支持的cuda版本<=最高cuda版本,点击下载

    如下图所示。下载离线安装包
在这里插入图片描述

四.离线安装Pytorch

1.打开Anaconda Prompt,激活之前创建的Anaconda的深度学习环境

#第一步:打开Anaconda Prompt,输入下述代码,激活之前创建的Anaconda的深度学习环境
conda activate deeplearn

在这里插入图片描述

2.第二步:打开下载好的Pytorch离线包所在文件夹位置,复制该地址

在这里插入图片描述

#第三步:激活anaconda 的deeplearn环境,代码如下
conda activate deeplearn
#输入下述代码
#cd +复制的地址
#以我的为例
cd E:\联想浏览器\data
#接着输入 E:,就是你的下载文件所在那个盘
E:
#结果如下图

在这里插入图片描述

#输入下述代码进行离线安装
#conda install --offline 下载的文件的名字
#切记!切记!一定要包含文件的后缀,如.bz2
conda install --offline pytorch-1.12.1-py3.9_cuda11.3_cudnn8_0.tar.bz2
#安装完成如下图所示

(ps:如果你需要安装其它包,也可以用这个方法)
在这里插入图片描述

#先进入python的交互模式
python
#接着导入pytorch
import torch
#结果发现会报错,如下图
#退出python交互模式
exit()

在这里插入图片描述
在这里插入图片描述

#我们需要安装numpy包
#在这之前对conda进行换源,输入以下代码
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
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
#直接输入conda install numpy
conda install numpy
#接着输入y,并回车,如下图所示:耐心等待既可。

在这里插入图片描述

五.检查是否安装成功

#在deeplearn环境下输入conda list
conda list
#结果如下图

在这里插入图片描述

#接着检查cuda,cudnn版本
#首先进入python的交互模式
#python交互模式,直接输入python即可进入
python
#接着输入下述代码
import torch #导入pytorch库
print(torch.cuda.is_available()) #查看是否有cuda
print(torch.backends.cudnn.is_available()) #查看是否有cudnn
print(torch.cuda_version) #打印cuda的版本
print(torch.backends.cudnn.version()) #打印cudnn的版本
#结果如下图

在这里插入图片描述

发现安装的cuda和cudnn版本与下载的离线包一样,至此就安装完成了,可以进行自己的深度学习了!

要在Ubuntu系统上安装CUDAcuDNN,可以按照以下步骤进行操作: 1. 首先,确保你的显卡驱动已经安装。你可以使用以下命令更新软件包列表并安装CUDA驱动: ``` sudo apt-get update sudo apt-get install cuda-drivers ``` 2. 接下来,你需要下载并安装CUDA Toolkit。可以使用以下命令下载CUDA安装文件: ``` chmod +x cuda_9.0.176_384.81_linux.run sudo ./cuda_9.0.176_384.81_linux.run --override ``` 注意,这里的安装命令是基于引用中提供的CUDA安装文件。 3. 安装CUDA Toolkit后,你可以继续安装cuDNNcuDNN是一个用于深度学习的加速库。你可以根据你使用的CUDA版本下载对应的cuDNN版本,并按照官方文档提供的步骤进行安装。 4. 安装cuDNN后,你可以继续安装其他需要的深度学习框架,比如PyTorch。你可以参考引用中提到的方法来安装PyTorch及其依赖库。 请注意,以上步骤仅供参考,具体操作还需要根据你的系统需求进行相应的调整。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [Ubuntu系统安装cudncudnn](https://blog.youkuaiyun.com/aoguren/article/details/79960025)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [如何在ubuntu系统上安装多个cudn环境(及对应cudnnpytorch 、torchvision安装方法 )](https://blog.youkuaiyun.com/qq_44776613/article/details/116997774)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ymchuangke

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值