当前位置:首页 » 《资源分享》 » 正文

深度学习环境的配置

7 人参与  2024年10月12日 10:01  分类 : 《资源分享》  评论

点击全文阅读


介绍:关于深度学习的框架有很多,比如国外的Pytorch,TensorFlow,Keras等等,国内的话例如百度的PaddlePaddle,华为的MindSpore。

我们学习的以Pytorch为主,因为目前Pytorch是主流趋势,pytorch是动态框架,tensorflow是静态框架,动态框架可以根据具体计算情况进行改变,并且pytorch对python十分友好,更加python化的框架。

而tensorflow的话更加像独立的语言和框架,不过TensorFlow因为出现比较早,库比较完善,在部署在开发板方面比较有优势,比如tflite量化之后可以在openmv和openart运行。

而对于我们比较友好的就是PaddlePaddle框架,可以多接触接触,因为落地非常快,并且教程也比较完善,并且官网上还会送一定量的云算力,国内的文心一言也是百度的,也是非常不错的。学习链接:飞桨PaddlePaddle-源于产业实践的开源深度学习平台

废话不多说,开始配置环境!

一.配置CUDA和CUDNN

1.确定是否独显

确定自己的电脑是否有独显,也可以通俗的说是否有英伟达的GPU,可以通过打开任务管理器然后查看自己的显卡,我的电脑是3050,你们估计都4060起步了吧。

如果没有独显的同学可以跳过配置CUDA和CUDNN这一步。

在这里插入图片描述

2.确定与自己电脑显卡相匹配的CUDA

CUDA简单的说就是“GPU通用计算”构建的运算平台,可以加快计算,大大提高模型训练和推理速度,有独显和没独显的速度根本不是一个量级的。

首先在cmd中输入

nvidia-smi

image-20240623005845758

以上显示的CUDA Version就是支持CUDA的版本,或者可以通过以上的驱动版本查找支持的CUDA版本

image-20240623011133356

3.下载和安装CUDA

如果有同学是3050,3060,3070的显卡可以直接到我的网盘里下载CUDA11.2和对应的CUDNN,40系列我不确定是否适配

[CUDA网盘链接](https://pan.baidu.com/s/1KhL7GBSO42A2XA9b_0hApw 提取码:67wp)

[CUDNN网盘链接](https://pan.baidu.com/s/17YQDM03XyKjZtFYPzR1v7w 提取码:be07)

确定好CUDA环境后然后到官网上进行下载对应的版本 CUDA下载链接

image-20240623011714435

信息填好之后点Download即可

CUDA下载完成后进行安装,安装选项选择精简即可,当然也可以选择自定义,有些驱动程序组件可能后面面还是会用到,如果Other components和Driver components的新版本比当前版本低的话就不用勾选,否则会导致安装失败。

安装路径建议全部默认即可,因为有时候我之前遇到过有时候编译的时候涉及到CUDA的话他默认的路径是在默认路径下查找的,改起来特别麻烦,所以C盘的压力不就上来了吗?

4.配置CUDNN

CUDNN是基于CUDA的深度学习加速库,可以简化和优化深度学习算法的开发和运行,CUDNN需要安装在CUDA的安装目录下,相当于CUDA的一个补丁。

首先去官网上下载:CUDNN下载链接

image-20240623123401240

如果CUDA是12点几的话就选择12.x的,如果CUDA是11点几的话就选择11.x的,选择对应的即可

下载过后应该是一个压缩包,解压之后文件夹下面应该是这三个文件夹

image-20240623123718744

然后找到你刚才所选择的CUDA下载路径,我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

在这里插入图片描述

然后CUDNN文件夹里面的bin,include和lib文件夹剪切或者赋值到v11.x的目录下,它会自动的将剪切的CUDNN文件夹下面的dll等文件放置对应的CUDA文件夹里面,CUDNN就相当于是扩展或者说是补丁。

然后打开其中bin文件内部是否有这些扩展,如果有则成功放置

image-20240623124454087

5.配置环境变量

如果安装选项选择的是精简的话,直接跳过这步,精简会自动帮你配置。

如果是选择自定义的话:那么需要任务栏搜索编辑环境变量

(1)在系统环境变量中添加以下变量

image-20240623123029749

(2)在Path中添加以下环境变量

image-20240623131132026

6.检验是否安装成功

进入cmd后输入

nvcc -V,注意大写

image-20240623131306318

如果出现以下输出则CUDA和CUDNN成功安装,如果没有的话大概率是环境变量没有配置好

二.安装Pytorch

1.新建一个python环境

因为不同的框架之间所用到的库可能版本可能有些冲突,比如numpy库等等,所以最好是新建一个环境,这里我就使用之前讲过的conda新建环境了。

还没有用conda的可以看看以下教程:conda安装

教程中Anaconda镜像链接:Anaconda镜像链接

进入Anaconda Prompt终端运行以下命令:

conda create -n pytorch python=3.9

2.安装pytorch

然后进入pytorch官网查找对应的命令行

Previous PyTorch Versions | PyTorch

如果没有独显只有CPU的话就选择,选择对应下载的torch版本然后选择最下面的命令行

image-20240623132629270

如果有独显的就选择选择对应的CUDA版本然后复制命令行

选择conda和pip都可,建议pip,conda可能有时候会下载失败

prompt后进入刚才新建的环境后,运行刚才复制的命令行,等待下载即可,包有点大,需要等一会,可以切换源

3.验证是否Pytorch是否配置成功

先进入刚才的pytorch环境

运行以下命令

image-20240623134100411

如果只有CPU的同学,第一个命令出现版本号即可

如果需要配置CUDA,运行第三个命令为True,那么恭喜至此Pytorch+CUDA环境配置成功

以上就是Pytorch配置的全部内容,感谢阅读,如有什么错误,希望能够指正!!!


点击全文阅读


本文链接:http://zhangshiyu.com/post/170891.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1