搭建yolov4环境踩坑

这个Up主很良心，做了很多相关的dl项目的解析。

源码地址：https://github.com/bubbliiiing/yolov4-pytorch
博客地址：https://blog.csdn.net/weixin_44791964/article/details/106214657

最近学习yolov系列，搭建环境

看了一下公司电脑配置 显卡竟然是GTX 1060 3G，想着可以训练一下模型

但公司的电脑是win7的系统，搭建cuda环境也是很多问题，最后也没办法重新装了个win10的系统

去英伟达(Nvidia)官网下载了相对最新的驱动

安装好cuda 后 才能使用nvcc --version 查看到的才是cuda版本，而通过NVIDIA控制面板看到的".dll"文件只是当前驱动支持cuda的最高版本而不是cuda版本

 

环境搭建好后，可以通过 

nvidia -smi

查看有哪些可用的gpu

 可通过

import torch
print(torch.cuda.is_available())

判断pytorch下是否配置好了cuda，判断gpu是否可用