在用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. 参考链接:
更多推荐
warnings.warn(‘User provided device_type of ‘cuda‘, but CUDA is not available)
发布评论