warnings.warn(‘User provided device

在用yolov5训练用户自定义数据集时运行过程中报错：

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

1. 产生原因：

原因是运行的环境与 torch 的版本不匹配

附：如何检测torch版本是否正确可用：

方法1：输入如下命令查看硬件设备

nvidia-smi

输出显示如下

然后在 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 版本如图所示。

复制安装的pip 命令进行安装，如果安装速度慢可以参考链接【】用国内的源进行下载。

重新安装合适版本的 torch 后，问题解决！

3. 注意：

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

4. 参考链接：