学无先后，达者为师

网站首页编程语言正文

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-05-06 C++提取文件名与提取XML文件的方法详解_C 语言
2023-01-01 C++ Boost.Signals2信号/槽概念_C 语言
2023-01-17 解读python cvxpy下SDP问题编程_python
2023-09-18 Mybatis新增数据，存在就更新，不存在就添加
2022-11-02 解析C/C++ Capstone 引擎源码编译问题_C 语言
2022-05-15 Python matplotlib seaborn绘图教程详解_python
2022-11-09 Flutter 异步编程之单线程下异步模型图文示例详解_Android
2022-05-09 .NET中常见的加解密算法详解_实用技巧

栏目分类

最近更新