【赛博炼丹】深度学习软件安装及环境配置:Anaconda、CUDA、cuDNN、PyTorch及PyCharm

前言

建议大伙自己建一个文件夹(不能有中文),专门放深度学习的软件,后续添加环境,比较方便。

1.安装Anaconda

1.1下载Anaconda

Anaconda官网:https://www.anaconda.com
清华大学镜像网站:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

安装Anaconda时,可以选择是否安装Python,不同的Anaconda版本自带的Python版本不同(也可以选择不安装Python),我安装的Anaconda版本为2021.11,所携带的Python版本为3.9。

清华大学镜像网站有Anaconda的往届版本,按需下载对应版本。

1.2安装Anaconda

安装包下载完成后,开始安装。

其他步骤均为默认,该步的安装类型选为All Users。

更改安装位置

第二个打勾

安装完成后,win+r,输入cmd,输入python -V,可以查看到Python版本为Anaconda2021.11自带的3.9版本。

1.3添加环境变量

搜索框中输入“查看高级系统设置”,一般输入“高级”二字就会出来,打开“查看高级系统设置”后,选择“环境变量”。

双击“Path”,选择“新建”,添加的环境变量名称为:安装路径+文件位置。

我是在D盘的DLSE文件夹中新建了anaconda这个文件夹,因此添加的四个环境变量如下:

D:\DLSE\Anaconda
D:\DLSE\Anaconda\Scripts
D:\DLSE\Anaconda\Library\mingw-w64\bin
D:\DLSE\Anaconda\Library\bin

1.4验证安装是否成功

启动Anaconda Navigator,顺利打开即安装成功。

2.安装NVIDIA显卡驱动

进入NVIDIA官网:NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

立刻下载GeForce驱动程序

注册一个NVIDIA账号,登录后,该程序能自动识别你电脑显卡驱动,然后更新显卡驱动。

安装完成后,在anaconda prompt或cmd中输入nvidia-smi,可以得到自己的显卡信息。

3.安装CUDA和cuDNN

3.1安装CUDA

3.1.1下载CUDA

上一步中,在Anaconda Prompt中输入nvidia-smi得到显卡的CUDA版本。

我的CUDA版本为12.6,那么所下载的CUDA不可高过这个版本。

N卡官网CUDA Toolkit Archive | NVIDIA Developer

选择合适版本,点击CUDA Toolkit 版本号,进入下载界面。

根据自己电脑,选择合适的配置,安装方式选“exe(local)”,点击Download。

3.1.2安装CUDA

安装CUDA时,第一次会让设置临时解压目录,第二次会让设置安装目录。

临时解压路径,建议默认即可,也可以自定义。安装结束后,临时解压文件夹会自动删除。

 

解压完成后,会自动打开安装程序,第四步的“安装”更改一下路径,其他步骤均为默认。

3.1.3验证安装是否成功

win+r键运行cmd,输入nvcc --version 即可查看版本号;

输入set cuda,可以查看 CUDA 设置的环境变量。

3.2安装cuDNN

3.2.1下载cuDNN

打开该网站cuDNN Archive | NVIDIA Developer,查看适配的 cuDNN。

选择跟自己的CUDA版本适配的cuDNN版本

我安装的是CUDA12.6,因此我选择cuDNN v8.9.7

下载完成后是一个压缩包

3.2.2安装cuDNN

将3.2.1所下载的安装包解压后得到以下文件:

CUDA的文件夹中,也有bin、include、lib这三个文件夹。

我们需要做的,便是将cuDNN三个文件中的内容复制到CUDA对应的文件夹中,由于我已经复制过,因此出现的是替换。

3.2.3添加环境变量

依旧是在Path中添加

4.PyTorch安装

4.1创建虚拟环境

在此之前,先添加清华镜像网站,Prompt中输入如下代码:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

在Prompt中创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下,我就是创建了一个名字叫yolov8,python是3.10版本的环境,在prompt中输入以下命令:

conda create -n yolov8 python=3.10

安装过程中出现Proceed([y]/n)?时,输入y,按回车键,继续进行安装。

安装完成后输入以下代码,进入yolov8虚拟环境:

conda activate yolov8

4.2Pip法安装PyTorch

进入PyTorch

选择Pip、Python、合适的CUDA版本,Prompt中运行“Run this Command”中的代码。

开始安装

4.3验证安装是否成功

激活对应的虚拟环境(你安装Pytorch的虚拟环境)

conda activate yolov8

输入pip list或者conda list,看有没有pytorch或者torch

分别输入以下代码:

python
import torch
torch.cuda.is_available()

import torch以后回车无error,第二行指令返回的是true就说明PyTorch安装成功。

5.PyCharm安装及配置

5.1安装PyCharm社区版

进入PyCharm官网,点击右上角的“Download”PyCharm: the Python IDE for data science and web development (jetbrains.com)

页面往下拉,下载社区版的PyCharm,专业版要钱,社区版免费,功能少点儿,但也够用了。

都选,记得更改安装路径。

5.2PyCharm配置

打开PyCharm后,新建项目,项目名称和位置自行修改,解释器类型选择“自定义环境”,选择现有环境,Conda类型,选择你的Anaconda安装路径下的conda.bat,环境选择之前配置的虚拟环境,我的是yolov8,点击创建。

点击右下角的解释器,添加新的解释器,添加本地解释器。

选择conda环境,使用现有环境,选择之前配置的虚拟环境,我的是yolov8,点击确定。

5.3验证CUDA和cuDNN版本

右键项目文件夹,新建Python文件,文件名随意。

输入以下代码:

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

运行结果如下,则证明安装成功。

至此,深度学习软件及环境配置完成。

参考文献

[1]2023最新pytorch安装(超详细版)-CSDN博客

[2]pycharm中添加conda环境所遇到的问题及解决方法汇总_pycharm无法加载conda环境-CSDN博客

[3]适合小白的超详细yolov8环境配置+实例运行教程,从零开始教你如何使用yolov8训练自己的数据集(Windows+conda+pycharm)-CSDN博客

[4]利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装---免额外安装CUDA和cudnn(适合小白的保姆级教学)_将python的版本切换为paddle环境中的python插件-CSDN博客

[5]深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】-CSDN博客

[6]新手保姆级GPU版本pytorch安装(win系统),CUDA和torch对应版本!_torch cuda-CSDN博客

[7]CUDA与cuDNN安装教程(超详细)-CSDN博客

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值