报错:AssertionError:torch not compiled with cuda enabled

这种情况就是自己的设置运行设备为GPU 但是本地的机子有没有GPU

有两种方法。一般会在初始化的时候就设置了是否用GPU

改成使用 CPU就行

第二种就是

在加载模型的时候 看到 load

torch.load(xxx, map_location=torch.device('cpu'))  # 这是终端报错提示我使用的格式

看到就变成上边的形式就行了

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • 面试题
    Swift1> Swift和OC的区别1.1> Swift没有地址/指针的概念1.2> 泛型1.3> 类型严谨 对...
    cosWriter阅读 11,161评论 1赞 32
  • iOS面试
    面向对象的三大特性：封装、继承、多态 OC内存管理 _strong 引用计数器来控制对象的生命周期。 _weak...
    运气不够技术凑阅读 1,153评论 0赞 10
  • Unity 优化归纳
    原文地址 http://www.fx114.net/qa-75-172454.aspx 使用Profiler工具...
    IongX阅读 5,981评论 1赞 11
  • ##《java虚拟机》汇总所有关键要点
    一 、java虚拟机底层结构详解 我们知道，一个JVM实例的行为不光是它自己的事，还涉及到它的子系统、存储区域、...
    葡萄喃喃呓语阅读 1,534评论 0赞 4
  • iOS启动时间优化【转】
    背景 一个项目做的时间长了，启动流程往往容易杂乱，库也用的越来越多，APP的启动时间也会慢慢变长。本次将针对iOS...
    酱油瓶2阅读 3,551评论 0赞 12