学无先后，达者为师

网站首页编程语言正文

python没有gpu,如何改用cpu跑代码_python

作者：孩纸D 更新时间： 2022-07-09 编程语言

目录

没有gpu,改用cpu跑代码

只需要将device设置为以下即可

Tensorflow-gpu禁用gpu设置(cpu与gpu速度对比)

禁用gpu设置
cpu与gpu对比

cpu
gpu

没有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

cpu

gpu

gpu

简单测试：gpu比cpu快5秒

原文链接：https://blog.csdn.net/weixin_42227243/article/details/124192982

上一篇：Android开发中Flutter组件实用技巧_Androi
下一篇：Pytorch从0实现Transformer的实践_pyth

相关推荐

2022-04-09 Python中缓存lru_cache的基本介绍和讲解_python
2022-03-24 C++ Primer的变量和基本类型详解_C 语言
2022-09-14 Python深入浅出分析enum枚举类_python
2022-06-12 C#泛型接口的协变和逆变_C#教程
2022-07-08 python open读取文件内容时的mode模式解析_python
2022-02-28 gyp info it worked if it ends with ok npm ERR 解决办法
2022-04-05 C#反射调用拓展类方法实例代码_C#教程
2022-06-12 python文件操作的基础详细讲解(write、read、readlines、readline)_p

栏目分类

最近更新