yolo v5 环境配置(gpu版本)

yolo v5 环境创建

运行YoloV5可选为CPU或GPU环境。默认CPU环境无需配置,GPU环境需要安装与显卡版本对应的cuda,
1、anaconda 环境创建 一个新的虚拟环境
yolo v5 环境配置(gpu版本)_第1张图片
2、激活切换环境,
去pytorch官网查询与CUDA相匹配的版本,运行相应命令 (https://pytorch.org/get-started/previous-versions/)
yolo v5 环境配置(gpu版本)_第2张图片
yolo v5 环境配置(gpu版本)_第3张图片

yolo v5 环境配置(gpu版本)_第4张图片
验证是否安装了gpu版本
yolo v5 环境配置(gpu版本)_第5张图片
3、下载yolo v5

4、pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple(注释掉torch等)
在这里插入图片描述
问题:
AttributeError: module ‘distutils’ has no attribute ‘version’
解决方案:
https://zhuanlan.zhihu.com/p/556704117
yolo v5 环境配置(gpu版本)_第6张图片
pip install setuptools==58.0.4 -i https://pypi.tuna.tsinghua.edu.cn/simple

yolo v5 环境配置(gpu版本)_第7张图片

RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place operation.
yolo v5 环境配置(gpu版本)_第8张图片
https://blog.csdn.net/hjxu2016/article/details/121303818

你可能感兴趣的:(深度学习,python,人工智能)