当前位置:   article > 正文

在ubuntu18.04下安装yolov5_ubuntu 配置yolov5环境

ubuntu 配置yolov5环境

在ubuntu18.04下安装yolov5

设备和系统:

1.电脑联想拯救者y7000,显卡GTX1650。
2.ubuntu版本18.04。

具体步骤

1.需要提前安装显卡驱动,具体安装过程在这里不加赘述,可以通过命令

nvidia-smi
  • 1

查看英伟达显卡驱动版本。
2.安装cuda,通过nvidia-smi查看显卡驱动对应的cuda版本。
cuda是nvidia推出的用于自家GPU的并行计算框架,也就是说cuda只能在nvidia的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥cuda的作用。
具体步骤参考链接:cuda安装教程
tip:在编写指令时注意cuda的版本。
3.安装Anaconda,注意下载和cuda对应的版本。
具体步骤参考链接:Anaconda安装教程
4.安装cudnn(此步可省略)
cudnn(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cudnn不是必须的,但是一般会采用这个加速库。
具体步骤参考链接:cudnn安装教程
5.配置pytorch虚拟环境
(1)配置环境变量供多用户使用Anaconda
(2)配置pytorch虚拟环境
注意:生成虚拟环境的路径应该为:

~/anaconda3/envs
  • 1

6.yolov5配置
7.相关模型验证
下载相关模型

tips:

1.查看显卡驱动版本
nvidia-smi
2.切换到pytorch虚拟环境下:
conda activate 虚拟环境(如pytorch)
3.下载依赖requirement.txt下载缓慢的解决方法
https://blog.csdn.net/hhxxtszj/article/details/128775991

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/971903
推荐阅读
相关标签
  

闽ICP备14008679号