在Windows10下,Git clone https://github.com/ultralytics/yolov5.git
然后運行:
報錯如下圖所示:pip install -r requirements.txt

UnicodeDecodeError: 'gbk' codec can't decode
問題原因:win10下特有的問題,win10的默認編碼方式是utf-8,而Python讀入文本文件的默認編碼方式是gbk
解決方式:找到read()函數(shù),顯示指定編碼方式。根據(jù)報錯信息,read()在dist.py文件的406行,所以打開該文件,加入“encoding="utf-8”,如下圖所示:

顯示加入編碼方式:“encoding="utf-8”

pip install -r requirements.txt 執(zhí)行成功

驗證PyTorch GPU版本

CUDA-10.2 PyTorch builds are no longer available for Windows
我的1080Ti 算力為6.1,從下面的三個表可以分析,還可以跟著PyTorch升級再戰(zhàn)幾年!

1080Ti 算力為6.1

Pascal構(gòu)架支持的顯卡驅(qū)動版本

CUDA與顯卡驅(qū)動版本的兼容性
YOLOv5的當前(3月28日)PyTorch版本:YOLOv5 ?? torch 1.10.0+cu102 CUDA:0 (Tesla V100 Volta構(gòu)架-SXM2-16GB, 16160MiB),看來跟我的1080Ti + torch1.10.0 + cu102差不多