基于pytorch的yolov5运行报错warnings.warn(‘User provided device_type of \‘cuda\‘, but CUDA is not available)

背景

今天运行基于pytorch的yolov5时报了warning

  • UserWarning: User provided device_type of ‘cuda’, but CUDA is not available. Disabling
  • warnings.warn(‘User provided device_type of ‘cuda’, but CUDA is not available. Disabling’)
    报错
  • 检验cuda和pytorch是否兼容

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

    我的输出结果如下,可以看出我的pytorch和cuda版本不兼容

    解决方法

    1. 打开pytorch官网 pytorch官网地址
    2. 按照自己电脑的配置选择,我是cuda10.2,windows电脑,安装的anaconda按下图选择
    3. 打开Anaconda Prompt,激活对应的环境
      激活‘XiGua’环境
    4. 将pytorch官网给出的命令复制上,pip3改为pip(根据自己的pip版本)
    5. 安装成功,显示True,torch为cuda加速版本

    来源:小白程序员成长记

    物联沃分享整理
    物联沃-IOTWORD物联网 » 基于pytorch的yolov5运行报错warnings.warn(‘User provided device_type of \‘cuda\‘, but CUDA is not available)

    发表评论