NvidiaRTX3070Ti/GTX960M深度学习环境搭建教程

本文分享了如何为深度学习搭建环境,包括检查NVIDIA驱动、安装CUDA11.1和CuDNN11.3、设置环境变量、安装Anaconda和PyTorch1.8.0,以及导入和测试代码的过程。适合拥有GTX960M或RTX3070Ti等配置的用户参考。
摘要由CSDN通过智能技术生成

距离上一篇发过的关于如何申请GitHub学生包的教程也过了一段时间,笔者在那段时间处于深度学习的入门状态,当时遇到的最大问题便是则是搭建运行环境。而环境搭不对,后续的工作便很难推进,因此这个话题是想给大家分享下在这个过程中我摸索出来的环境配置,以供参考。

在寻找环境的过程中笔者也参考了一些教程也挺有借鉴,连接如下,大家可以看看。
960M环境搭建配置
CUDA环境安装

我使用的配置有两台分别是:
1、GPU:GTX960M-4G显存一块
内存:16G
CPU:i7-6700HQ
2、GPU:RTX3070Ti-8G显存两块
内存:64G
CPU:i9-13900k

Pytorch环境:python=3.7,CUDA=11.1 CUdnn=11.3,pytorch=1.8.0,

需要注意的是以上两台用到的环境都是一样,因此如果你的电脑是这两配置中的任意一个都是可以用的,另外如果你的电脑性能是处于这两台之间的,笔者猜测也是可行的可以试试。

那么如何顺利搭建并成功运行环境呢,首先,我们需要先了解电脑中的显卡的驱动的版本,看看驱动是否是最新(以防万一,尽量升级最新版本),可以通过电脑win+R打开运行,然后在运行中输入并打开cmd,在cmd内输入nvidia-smi,看最上面的Driver Version。如果驱动不是是最新,可以按照连接教程的方法二来登录英伟达官网更新。

接着我们需要给自己的电脑上安装CUDA和CUdnn,在这里面我安装的CUDA11.1和CUdnn11.3( 笔者已经安装包上传至网盘内 如果你嫌下载麻烦可以网盘下载)。需要注意的是这两个在后续运行的神经网络时至关重要,如果安装不对,很容易导致出现cuda无法获得的报错。CUDA和CUdnn的安装过程可以参考下面这个视频教程:
CUDA及CUdnn安装教程
链接:https://pan.baidu.com/s/1l1bZ_hhwL47FFGZux0O-nA
提取码:2023

当按照教程安装完CUdnn后还需要将lib、bin、linclude文件夹添加至环境变量中。添加的教程可以参考以下连接:
文件添加环境变量方法
在这里插入图片描述

安装完后,接着开始下载安装anaconda,安装过程可以参考这个视频的教程:
conda安装
之后就是pytorch的安装,安装过程仍然参照这个视频的教程来,只将其中的pytorch下载版本换成1.8.0即可,具体如下:
在这里插入图片描述其他的则不变。

安装好pytorch后接下来就是安装pycharm和环境导入,安装过程参考这个视频即可。
pycharm及环境导入

当一切安装好后,可以去GitHub随便找一个深度学习代码试试,如果能跑起来就说明能行了(在跑代码的过程中可能会出现模块没有导入的情况,这个不同担心,在anaconda prompt中导入所缺失的模块即可,具体教程可以参考以下连接

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

量子屋

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

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

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

打赏作者

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

抵扣说明:

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

余额充值