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了