一、问题描述：
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了
标签：yolov5,记录,torch,pytorch,conda,版本,gpu,yolo5,cuda10.2 From： https://www.cnblogs.com/wjx-blog/p/17007733.html