在Windows10下,Git clone https://github.com/ultralytics/yolov5.git
然后運(yùn)行:
報(bào)錯(cuò)如下圖所示:pip install -r requirements.txt
UnicodeDecodeError: 'gbk' codec can't decode
問題原因:win10下特有的問題盹牧,win10的默認(rèn)編碼方式是utf-8泣棋,而Python讀入文本文件的默認(rèn)編碼方式是gbk
解決方式:找到read()函數(shù)咽筋,顯示指定編碼方式距误。根據(jù)報(bào)錯(cuò)信息宪拥,read()在dist.py文件的406行信夫,所以打開該文件窃蹋,加入“encoding="utf-8”,如下圖所示:
顯示加入編碼方式:“encoding="utf-8”
pip install -r requirements.txt 執(zhí)行成功
驗(yàn)證PyTorch GPU版本
CUDA-10.2 PyTorch builds are no longer available for Windows
我的1080Ti 算力為6.1警没,從下面的三個(gè)表可以分析,還可以跟著PyTorch升級再戰(zhàn)幾年振湾!
1080Ti 算力為6.1
Pascal構(gòu)架支持的顯卡驅(qū)動版本
CUDA與顯卡驅(qū)動版本的兼容性
YOLOv5的當(dāng)前(3月28日)PyTorch版本:YOLOv5 ?? torch 1.10.0+cu102 CUDA:0 (Tesla V100 Volta構(gòu)架-SXM2-16GB, 16160MiB)杀迹,看來跟我的1080Ti + torch1.10.0 + cu102差不多