千家信息网

python没有gpu怎么改用cpu跑代码

发表于:2024-10-22 作者:千家信息网编辑
千家信息网最后更新 2024年10月22日,这篇文章主要介绍"python没有gpu怎么改用cpu跑代码"的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇"python没有gpu怎么改用cpu跑代码"文章能帮
千家信息网最后更新 2024年10月22日python没有gpu怎么改用cpu跑代码

这篇文章主要介绍"python没有gpu怎么改用cpu跑代码"的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇"python没有gpu怎么改用cpu跑代码"文章能帮助大家解决问题。

没有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 osos.environ['CUDA_VISIBLE_DEVICES'] = '-1'

cpu与gpu对比

显卡:GTX 1066

cpu

gpu

简单测试:gpu比cpu快5秒

关于"python没有gpu怎么改用cpu跑代码"的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识,可以关注行业资讯频道,小编每天都会为大家更新不同的知识点。

0