当你能够搜索3060时,我能感同身受到你能有多绝望!
这篇文章受众很小,仅支持使用NVIDIA 3060 显卡来使用yolov5训练模型的用户
***NIADIA 3060 仅支持cuda 11.1 以上版本***
(这是我在搜索中找到的一条信息,原地址给忘记了,所以不能给出转载地址,我之前用的是cuda10.2版本,在使用yolov5时不能调用GPU)
因网速受限,Aconda、cuda、cuDNN 都是离线下载,再用pip按安装的
我选择的是cuda11.5、支持cuda11.5的pytorch1.11
关键词:
yolov5
NVIDIA 3060
Python=3.9
接下来说一下步骤:
1、anaconda安装和虚拟环境创建这个就不在赘述
2、下载cuda(手动下载)