ubuntu20部署yolov5环境 详细版 pytorch

ubuntu20部署yolov5环境 详细版 pytorch

提示：写完文章后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

提示：这里可以添加本文要记录的大概内容：

在ubuntu20下（win也差不多）部署yolov5环境，训练自己的模型 pytorch

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、下载Nvidia驱动

查找自己电脑是否有nvidia显卡，找到对应的驱动进行安装
ubuntu可以使用nvidia-smi验证是否安装好
显示以下表示成功

二、安装cuda

nvidia-smi查看驱动，例如以上的12.2，说明最好是下载12.2的CUDA，比他低一点也可以

官网找到适合自己的版本下载

https://developer.nvidia.com/cuda-toolkit-archive

已经安装驱动就把driver取消掉，光标移到对应位置enter就行
然后进行intall

加入环境变量

 gedit ~/.bashrc 
 在最下面加入
 export PATH="/usr/local/cuda-12.2/bin:$PATH"
 export LD_LIBRARY_PATH="/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH"

刷新一下
source ~/.bashrc

输入