深度学习环境安装配置


一、概述

1、操作系统:win10

2、显卡:3080单卡

3、Anaconda:Python3.8版

4、visual studio:community 2019:安装cuda必须要安装visual studio;

5、深度学习框架:pytorch

6、客户端IDE:PyCharm

7、yolov5目标检测

 

二、visual studio community 2019安装

 

安装cuda必须安装visual studio(Microsoft C++ Build Tools)

 

下载位置:https://visualstudio.microsoft.com/zh-hans/

选择community 2019下载

 

选择C++的桌面开发

 

 

选择Visual Studio扩展开发

 

安装文件较大,根据实际情况选择安装位置

 

开始安装

 

A few years later 安装结束后重启。

 

三、安装cuda和cudnn

 

1、先查看本机安装的显卡信息

win+r—cmd—输入nvidia-smi查看Driver Version为466.81,CUDA Version是11.3

※ 这里的CUDA Version是展示系统可支持到的最新版本,而不是“该系统已经安装了此版本”。

 

2、查看CUDA与Driver Version对应关系,可见支持CUDA 11.3.1

官网地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

 

3、或者在本地验证:进入NVIDIA控制面板——帮助——系统信息——组件,看到RTX 3080最高支持CUDA 11.3.122版本

 

 

4、前三步的目的就是确定本机合适的cuda版本,这个版本和显卡驱动版本强关联。

确定版本:安装cuda11.3.1

 

去官网下载:https://developer.nvidia.com/cuda-toolkit-archive

2.7G+,建议使用迅雷下载。

 

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

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

安装目录,建议默认即可;

注意:临时解压目录千万不要和cuda的安装路径设置成一样的,否则安装结束,会找不到安装目录的!!!

选择精简安装(懒,不想安装完缺这缺那的)

 

安装完成后,配置cuda的环境变量查看(系统自动追加无需手动配置)

 

5、命令行测试是否安装成功;

nvcc -V

 

6、cudnn下载和安装

官网下载:https://developer.nvidia.com/rdp/cudnn-archive

下载需要登录,没有会员注册一下会员。

大小:740M+

 

打开cudnn文件夹,把cudnn中bin、include、lib/x64这三个文件夹里面的文件,分别全部复制到对应的cuda的bin、include、lib/x64文件夹内。

 

重启电脑。

 

7、检查安装是否成功

通过NVIDIA提供的 deviceQuery.exe 和 bandwidthTest.exe 来查看GPU的状态,两者均在安装目录的 extras\demo_suite文件夹中

 

deviceQuery.exe

 

bandwidthTest.exe

 

四、创建虚拟环境

 推荐使用Anaconda,不推荐直接安装Python.exe。因为后面接触的开源项目可能非常多,有些安装同一个环境下会出现版本冲突,Anaconda具有隔隔离效果,所以使用。

 

1、创建一个名为yolo、python版本为3.8的虚拟环境。

conda create -n yolo python=3.8

 

2、激活创建好的环境

activate yolo

 

3、安装pytorch,上pytorch官网查询安装命令

https://pytorch.org/get-started/locally/

 

根据自身的CUDA型号,选择最接近的CUDA11.1版本。

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c conda-forge

 

上面能否成功看人品。如果安装失败多试几次,如果再失败,尝试国内源,不过有些版本可能不支持。

 

4、验证是否成功

import torch

import torchvision

print(torch.cuda.is_available())

 

返回True表示CUDA配置成功~!撒花 (●'◡'●)~

 

五、配置yolo

 

1、github找到yolov5的地址,通过gitclone将代码克隆到本地

 

2、安装PyCharm作为IDE,安装后打开本地的该项目目录,并导入python解释器地址。

 

3、通过命令行工具进入工程目录,安装依赖包

pip install -r requirements.txt

 

4、打开PyCharm,执行detect.py,去项目下 /runs 目录查看输入结果。正常输出代表环境配置成功。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM