当前位置:   article > 正文

深度学习入门,配置pytorch环境

深度学习入门,配置pytorch环境

大家在入门深度学习之前,必不可少的步骤一定是配环境咯!!!下面布灵布灵磊少为大家讲述一下如何配置pytorch的步骤

pytorch一般分为cpu版本和gpu版本,一般cpu跑深度学习代码不太现实,电脑很容易坏,所以接下来布灵布灵磊少带给大家保姆级安装pytorch教程。

cpu版本我就不在此叙述了,懂得都懂。我给大家带来gpu版本,在安装之前先看一下自己电脑配置,是否符合gpu训练的条件,布灵布灵磊少原先的电脑为DELL,显卡最低配置为NVIDIA GeForce GTX 1050(显卡中下水平),不符合训练条件的就别看下去了。

 打开你的NVIDIA控制面板,打开帮助下的系统信息,点击组件,查看你的设备,若你的NVIDIA CUDA版本低于11.3,就别看下去了。

 49fddc2a05a8408d8ea43981f59cb9d1.png

 众所周知,系统cuda版本是向下兼容的,举例:本系统cuda为12.0,则可以安装12.0以下的cuda toolkit。本人为大家以11.3的cuda toolkit作为示范,大家可以去查资料下载,也可以从我的百度网盘下载下图的两个文件,分别是cuda toolkit11.3和对应版本的cudnn

链接:https://pan.baidu.com/s/15VgaB-sJDc-8zN1I_SyrLA?pwd=1111 
提取码:1111

ba18ef419ee547979dae155a5e23e11e.png

CUDA11.3安装

接下来参考这篇博主☞源仔,依照博主的步骤安装如下:

第一步:下载cuda_11.3.0_465.89_win10.exe软件后,双击安装,弹出安装保存路径,选择默认路径(高手可以自定义路径)后等待安装。(由于自己安装时没有截图,借用博主☞源仔安装cuda10.2版本的图片)

213ebea9fa8841089e0ecb243648561d.pngbe39ebe6b9134fd5a988a535e5f81852.png

第二步:提取完成后弹出安装界面

eac098bf57a64c5e8c72d4639fdb27a9.png

第三步:同意并继续

8979bacf0dbe4267bb7fd75f378323b3.png

第四步:小白尽量选择自定义,因为精简会将显卡驱动重装

1b685059b67a4008be2bd7b24ea31eb5.png

第五步:取消Driver components勾选,另外都勾上

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

闽ICP备14008679号