python没有gpu,如何改用cpu跑代码
短信预约 -IT技能 免费直播动态提醒
没有gpu,改用cpu跑代码
以为很麻烦的修改,很大动干戈的改动,以及无止尽的报错,结果,就改动代码中的device就OK了,惊不惊喜,意不意外!?
也就是说,
只需要将device设置为以下即可
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu"))
另外也不需要全如此设置,只需查看报错处调用device的来源,在修改torch.device()括号里面的内容就可。
Tensorflow-gpu禁用gpu设置(cpu与gpu速度对比)
禁用gpu设置
# 在import tensorflow之前
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
cpu与gpu对比
显卡:GTX 1066
cpu
gpu
简单测试:gpu比cpu快5秒
以上为个人经验,希望能给大家一个参考,也希望大家多多支持编程网。
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341