这篇文章主要介绍“python没有gpu怎么改用cpu跑代码”的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇“python没有gpu怎么改用cpu跑代码”文章能帮助大家解决问题。
以为很麻烦的修改,很大动干戈的改动,以及无止尽的报错,结果,就改动代码中的device就OK了,惊不惊喜,意不意外!?
也就是说,
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu"))
另外也不需要全如此设置,只需查看报错处调用device的来源,在修改torch.device()括号里面的内容就可。
# 在import tensorflow之前 import os os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
显卡:GTX 1066
简单测试:gpu比cpu快5秒
关于“python没有gpu怎么改用cpu跑代码”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识,可以关注亿速云行业资讯频道,小编每天都会为大家更新不同的知识点。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。