Pytorch-Yolov5從0運行Bug解决：

目錄超鏈接：

• Q: 需要安裝對應版本的 pytorch 、 torchvision 、opencv 等一系列的庫, 從哪裏安裝呢？（如何完美的換源）
• Q: import cv2 出現 libGL.so.1: cannot open shared object file: No such file or directory
• Q:AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

流程： docker 創建好鏡像後, 鏡像中安裝anaconda後，創建好 yolov5的環境

Q: 需要安裝對應版本的 pytorch 、 torchvision 、opencv 等一系列的庫, 從哪裏安裝呢？（如何完美的換源）

Details：但是Anacodna 默認的源采用的是國外的，會導致我們安裝東西特別慢，pip (python包管理工具的源也是國外的，需要我們換源、或者安裝的時候，臨時換源)

解决方法：

1. anaconda 添加源的時候，一定要注意寫 http 不要寫 https

# 在終端命令行執行該命令
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

執行完之後，你會發現在 home 目錄下多了一個 .condarc 的文件

• vim ~/.condarc
  內容如下：

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

至此，換源成功，可以根據官網的 pytorch 安裝命令去安裝。

• 切記， 去掉末尾的 -c Ptyorch ( -c, 是指 -channel 從Pytorch 官方的網址安裝，特別慢，有很大概率會失敗)
• Pytorch 官網：
  • https://pytorch.org/
  • 老版本：https://pytorch.org/get-started/previous-versions/

2. pip 換源

#清華：
https://pypi.tuna.tsinghua.edu.cn/simple
#阿裏雲：
http://mirrors.aliyun.com/pypi/simple/
#豆瓣：
http://pypi.douban.com/simple/

• yolov5下的安裝命令（直接拉取並且安裝好 yolov5 對應版本下的所有庫）：

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple –-trusted-host pypi.douban.com

Q：import cv2 出現 libGL.so.1: cannot open shared object file: No such file or directory

解决方法：

pip install opencv-python-headless

參考自：https://blog.csdn.net/qq_50195602/article/details/124188467

這比特仁兄講解的更為透徹，甚至講了如何在 dockerfile 中增加安裝指令，去規避這個問題。

Q:AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

描述：我推理 yolov5 示範例子圖像的時候產生的錯誤。

原因： torch 和 torchvision 版本不匹配導致的；

解决方法：

1. 安裝匹配的版本
2. 更改 upsample.py 中的代碼， 去掉函數那個參數

參考： https://blog.csdn.net/qq_40280673/article/details/125095353