yolo5使用gpu时遇到的问题记录

一、问题描述：

1、训练的时候提示不支持gpu

2、使用如下命令检查为False

import torch
torch.cuda.is_available()

二、原因：

pytorch版本的问题 

三、解决办法：

 重新安装符合的版本，参考的是：

torch.cuda.is_available() 解决方案 - 知乎 (zhihu.com)

 

 

把原来的版本删掉，然后安装新版本：

我是直接把环境删了，然后重新建后再做的：

删掉原来的环境再重新创建

conda env remove -n yolov5_cuda10.2

conda create -n yolov5_cuda10.2

python==3.9 conda activate yolov5_cuda10.2

安装符合条件的1.10.1版本的pytorch

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2

四、结果

成功地显示为True了