使用pytorch GPU训练YOLOv5遇到的坑

1. import torch

torch.cuda.is_available() 返回False， 发现需要使用torch对应的GPU版本：

https://download.pytorch.org/whl/torch_stable.html

        查看版本：

         import  torch

         torch.__version__

        import   torchvision

        torchvision.__version__

注意下面红色的CUDA 11.1版本和python版本3.7。

cu111/torch-1.8.1%2Bcu111-cp37-cp37m-win_amd64.whl

torchvision-0.9.1+cu111-cp37-cp37m-win_amd64.whl

https://developer.download.nvidia.cn/compute/cuda/11.1.0/local_installers/cuda_11.1.0_456.43_win10.exe

2. 出现了如下错误：

Error loading “\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll“ or one of its dependencies.

下载VC_redist.x64.exe，安装相关的dll文件即可。

pip install intel-openmp

另外出现页面分配不足时，需要调整硬盘对应的虚存：

3. 训练时出现下面错误， 学要修改nw=0：

    ForkingPickler(file, protocol).dump(obj)

BrokenPipeError: [Errno 32] Broken pipe

create_dataloader

nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, workers])  # number of workers

nw = 0