服务器环境搭建Yolov5的环境并简单测试
在为服务器搭建环境时,已提前下载好相应的CUDA版本;仅需下载PyTorch及其配置文件中指定的需求项即可.
1.创建环境
参考链接：搭建环境
1.1 创建虚拟环境
1.首先在本地环境下搭建一个环境，名字设为yolo5
conda create -n yolov5 python=3.7#创建环境
bash2.查看conda环境下是否有刚才创建的环境
conda env list
bash结果如下：
3.激活虚拟环境
conda activate yolov5#切换yolov5环境
bash1.2 配置虚拟环境
1.查看当前cuda的版本
nvcc -V
bash建议用的11.3的版本，因为其他版本我没有配置成功，就很玄学。
- 修改 CUDA 版本（若当前 CUDA 版本为 4.x 或其他非 5.x 版本，则无需操作）
若非当前 CUDA 版本为 5.x 系列，请您先获取 5.x 系列的 CUDA 版本，并按照以下命令行方式设置 CUDA 位置
vim ~/.bashrc
bash编辑修改cuda版本
官方提供的下载指引：下载指令
使用conda进行安装的包包括pytorch、torchvision、torchaudio，并指定版本为cudatoolkit=11.3，默认从pytorch官方仓库获取
检查pytorch和cuda是否匹配
import torch
torch.cuda.is_available()
torch.zeros(1).cuda()
bash- 获取requirement中的软件包
访问GitHub平台获取yolov5的源代码
yolov5源码
然后解压到本地文件，通过FileZilla或者Xshell上传到服务器
下载requirements.txt中要求的包，输入指令如下
conda activate yolov5 #激活环境
pip install -r requirements.txt #按要求下载包
bash到这里，环境配置就应该结束了。
2.测试环境是否搭建成功
参考链接：测试
输入以下指令测试
python detect.py --source ./data/images/bus.jpg --weights yolov5s.pt
bash
说明环境搭建成功，yolov5可以正常。
3.在这个过程中用到的相关指令
1.参考链接：相关指令
进入vim
vim ~/.bashrc
进入编辑模式
i
退出编辑模式
Esc
退出
输入 ：wq （ ：不可忽略，保存退出）
输入 ：q! （不保存退出）
2.参考链接：相关指令
新建虚拟环境
conda create -n env_name python=x.y
列出所有虚拟环境
conda env list
删除虚拟环境
conda remove -n envname --all
激活虚拟环境
source activate env_name
退出虚拟环境
source deactivate
4.在环境配置中遇到的问题
第一个遇到的问题是计算资源匮乏。经过查询资料后了解到：通常会指出 cuda、pytorch 和 GPU 的版本问题。经过尝试多个版本后发现 11.3 版本能够顺利运行。
在尝试下载PyTorch时遇到了问题。我成功地通过清华源获取了官方版本，并对配置文件进行了必要的调整以解决安装过程中的异常情况。参考链接：清华源
输入指令进入编辑界面
vim ~/.condarc
bash修改内容为
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bash
就可以正常下载了，而且速度非常快！！！