首先声明,我只是发出来记录一下。本人新手小白,看了好多教程,最后参考的炮哥的,搜索炮哥带你学即可。下面放上我的过程供大家参考
Anaconda+pytorch+paddle+pycharm安装深度学习环境
首先确认显卡驱动程序版本
Win+R cmd nvidia-smi
确认当前显卡驱动版本+最高支持的CUDA等级
2021.11.17
Anaconda官网Anaconda | Individual Edition
双击下载好的安装包,按步骤安装就行了
选择D盘,新建文件夹Anaconda3,避免以后C盘占用空间太多影响系统运行
我这里选择的是不打勾,手动添加环境变量。
配置环境变量,右击我的电脑-属性-右侧高级系统设置-环境变量
找到Anaconda的安装目录,依次添加如下几个目标路径到环境变量里
验证:
WIN+R CMD 输入conda,显示如下即成功
Pytorch的安装
点击Win键,Anaconda终端的Anaconda Prompt
基本命令:
查看全部环境conda env list
创建环境 conda create -n 环境名 python=版本
激活环境conda activate 环境名
退出当前环境conda deactivate 环境名
删除当前环境 conda remove -n环境名 --all
Pytorch的安装
首先创建一个虚拟环境conda create -n pytorch python=3.8
创建好了以后,就可以在这个虚拟环境安装一些后续需要使用的Python包了,激活刚刚创建的环境conda activate pytorch
给Pytorch安装路径换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
然后打开pytorch的官网PyTorch
,选定版本,复制红框范围内的文字到Anaconda执行
允许后等待安装就行了。
Paddle的安装同样如此,创建环境conda create -n paddle python=3.8
激活环境conda activate paddle
官网飞桨PaddlePaddle-源于产业实践的开源深度学习平台
等待安装即可
Pycharm选择的是2021版本,公众号软件智库有破解中文版的安装教程,,回复Pycharm即可,按照步骤安装。
打开新建一个工程
这里第一次新建工程需要下载一个python包,权限允许,等待下载就行了
右下角添加解释器,添加我们在Anaconda中安装的学习环境即可,首先选择Pytorch环境
创建一个.py文件
输入
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
输出
True
True
11.3
8.10(貌似是,只要有对应版本就行)
同样的在Paddle环境下创建一个脚本文件
输入
import paddle
print(paddle.utils.run_check())
输出
有下面这一行代码问题不大
The installed Paddle is compiled with CUDNN 8.2, but CUDNN version in your machine is 8.1, which may cause serious incompatible bug. Please recompile or reinstall Paddle with compatible CUDNN version
只要能成功输出PaddlePaddle is installed successfully即可
基本环境安装结束
Yolov5
www.github.com官网下载源码
更改host文件提升访问速度百度具体方法即可。
GitHub访问太慢,修改hosts文件提升访问速度!_益添的博客-CSDN博客_github hosts
下载的yolov5-5.0版本,下载好解压到桌面,用Pycharm打开,添加之前的pytorch环境,打开requestments.txt文件在底下终端命令行运行pip install –r requestments.txt安装所需要的包就行了
需要注意的是一个pycocotools 包,会提示缺少VC++14.0的支持,打开对应的网站,选择C++桌面开发,单个组件搜索build,选择适用于V142的C++生成工具,大概不到7个G,下载安装就行了 即可解决pycocotools安装失败问题
回到pycharm底部终端 pip install pycocotools 即可。
关于opcencv-python 问题 同样终端试一下pip install opencv-python
我第一次没成功,第二次成功了,具体原因不详
然后就是训练
目标检测---教你利用yolov5训练自己的目标检测模型_didiaopao的博客-CSDN博客
根据自己需要,更改目标位置,权重位置即可。具体还得自己摸索