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

cpu与gpu对比

显卡：GTX 1066

cpu

gpu

简单测试：gpu比cpu快5秒

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