Ubuntu20.04安装pytorch和yolov5环境

每台电脑的环境都有差异，出现的报错可能不尽相同，博客和视频仅供参考，希望能对大家有所帮助。

本文所用到的包都在云盘上可以下载
链接：https://pan.baidu.com/s/149EOhd4csxC_-QZJP1Jd9w 
提取码：0721

博客：https://blog.csdn.net/qq_64006507?spm=1011.2415.3001.5343
b站：https://space.bilibili.com/34693773?spm_id_from=333.1007.0.0
内容均在b站与博客上有演示

准备阶段：按照我上一篇博客安装好显卡驱动CUDA和CUDNN之后

安装pip：

sudo apt install python3-pip

下载历史版本pytorch：https://pytorch.org/get-started/previous-versions/

（主页面：https://pytorch.org/get-started/locally/)

找到跟你对应的CUDA版本安装

安装的版本建议不要超过你的CUDA版本，例如我的CUDA版本是11.4的，就选择11.3的来安装。

pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple/

（ -i https://pypi.tuna.tsinghua.edu.cn/simple/ 是换源）其他换源地址：

国内镜像地址：
阿里云
http://mirrors.aliyun.com/pypi/simple/

中国科技大学
https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣(douban)
http://pypi.douban.com/simple/

清华大学
https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学
http://pypi.mirrors.ustc.edu.cn/simple/

安装好以后去下载yolo5的包

！！！这边推荐使用我云盘上的YOLOV5的文件夹 里面的一些参数我都修改过的 导出来的onnx文件可直接给opencv读取，后面会演示！！！

网址：https://gitcode.net/mirrors/ultralytics/yolov5?utm_source=csdn_github_accelerator

切换分支，选择YOLOV5的版本（推荐选择V6.0或者V5.0的版本，安装起来方便，后面模型转换效果也好）

下载解压后进入该文件夹

pip3 install -r requirements.txt

如果报红了可以去根据提示查一下包的版本问题

可以pip3 uninstall 【包名】==【版本】#卸载

然后根据提示pip3 intall 【包名】==【版本】#安装

直到重新pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/后不再报错

测试

python3 detect.py --weights yolov5s.pt --source 0

若出现下面问题，将文件夹内的Arial.ttf文件复制到主目录的对应文件夹内，

回到主目录按Ctrl+H将隐藏文件显示出来，进入.config文件夹再进入Ultralytics文件夹，将Arial.ttf文件复制到这里，然后再回到yolo5的终端运行python3 detect.py –weights yolov5s.pt –source 0

若出现下述问题

进入下面路径的文件夹

回到主目录进入.local文件夹

找到upsampling.py文件并打开

在153行左右找到下面这句，删掉所选

删掉以后如下图，然后Ctrl+S保存退出

再回到yolov5的终端运行，成功显示