warnings.warn(‘User provided device_type of ‘cuda‘, but CUDA is not available)

在用yolov5训练用户自定义数据集时运行过程中报错:
warnings.warn(‘User provided device_type of ‘cuda‘, but CUDA is not available)
在这里插入图片描述

1. 产生原因:

原因是运行的环境与 torch 的版本不匹配
附:如何检测torch版本是否正确可用:
方法1:输入如下命令查看硬件设备

nvidia-smi

输出显示如下
warnings.warn(‘User provided device_type of ‘cuda‘, but CUDA is not available)_第1张图片
然后在 Pytorch 官网查看合适的cuda 版本。
方法2:
在当前环境下新建一个.py问价输入如下代码:

import torch

print(torch.__version__)
print(torch.cuda.is_available())

运行后的输出最后一行如果是

False

说明torch有问题,如果最后一行的输出是:

True

则torch 没有问题

2. 解决办法:

首先用以下命令查看当前环境下安装的所有包版本

pip list

主要查看torch和torchvision的版本
在这里插入图片描述
然后输入命令:

uninstall torch
uninstall torchvision 

将 torch 包彻底删除
然后在 Pytorch 下载 官网查看合适的cuda 版本如图所示。
warnings.warn(‘User provided device_type of ‘cuda‘, but CUDA is not available)_第2张图片
复制安装的pip 命令进行安装,如果安装速度慢可以参考链接【】用国内的源进行下载。
重新安装合适版本的 torch 后,问题解决!

3. 注意:

一定要找的合适版本的cuda 和torch,建议最好在虚拟环境中操作,以免造成其他影响。

4. 参考链接:

你可能感兴趣的:(yolov5训练自定义数据集,pytorch,深度学习,人工智能)