tensorflow只用CPU运行的方法

最新推荐文章于 2024-05-20 15:40:11 发布
转载 最新推荐文章于 2024-05-20 15:40:11 发布 · 4.2k 阅读 · 6 ·
CC 4.0 BY-SA版权
原文链接：https://blog.youkuaiyun.com/rookie_wei/article/details/93411794

本文介绍如何通过设置环境变量CUDA_VISIBLE_DEVICES为-1来禁用可见的GPU资源，确保Python脚本运行时不使用GPU加速。这对于没有GPU资源或希望避免GPU使用的场景非常有用。
部署运行你感兴趣的模型镜像

在要执行的Python文件首加上下面的代码即可， （把可见的GPU标记为-1）

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"

您可能感兴趣的与本文相关的镜像

Python3.9

Python3.9

Conda
Python

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

wzg2016
关注
Keras-tensorflow 下cpu-gpu相关问题
FREEDOM
10-25 471
特别注意版本问题 亲自实践过的版本 windows 10 tensorboard 1.8.0 tensorflow-gpu 1.8.0 pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.8.0 Keras 2.2.4 pip install -i https://pypi.mirrors.ustc.edu.cn/simpl...
tensorflow只用CPU运行
qingfengxd1的博客
11-02 903
import os os.environ["CUDA_VISIBLE_DEVICES"]="-1"
Tensorflow模型预测时，若用CPU计算，如何设置？ 并行计算-多核(multicore)，多线程(multi-thread)
郭云飞的专栏
07-05 1万+
在标准系统上通常有多个计算设备. TensorFlow 支持 CPU 和 GPU 这两种设备. 用指定字符串来标识这些设备. 比如: "/cpu:0": 机器中的 CPU "/gpu:0": 机器中的 GPU, 如果你有一个的话. "/gpu:1": 机器中的第二个 GPU, 以此类推... 在Tensorflow程序中，我们会经常看到 with tf.device("/cpu:0"): 这个语...
TensorFlow的CPU与GPU指定问题
GUEST_MTE的博客
02-02 1700
当同时安装CPU版本与GPU版本的TensorFlow时，默认是如果版本相同，默认是使用GPU版本的。
tensorflow 在指定CPU/GPU上运行方法
热门推荐
haima1998的专栏
07-20 2万+
转自：http://wiki.jikexueyuan.com/project/tensorflow-zh/how_tos/using_gpu.html 使用 GPUs 支持的设备 在一套标准的系统上通常有多个计算设备. TensorFlow 支持 CPU 和 GPU 这两种设备. 我们用指定字符串 strings 来标识这些设备. 比如: "/cpu:0": 机器中
Win10下Tensorflow-gpu使用cpu训练而不是gpu加速
weixin_43716328的博客
04-02 4650
Win10下Tensorflow-gpu使用cpu训练而不是gpu加速
深度学习CPU版本环境搭建（从anaconda-＞pycharm-＞tensorflow）
weixin_45427893的博客
09-23 4113
前言：其实当自己写下这个搭建过程的时候，自己已经碰壁很多了，所幸一路上还是有csdn博客论坛指导，特别鸣谢小艾童鞋熬夜写了一份安装指南给我，以及廖大师兄的悉心引导。所以自己也想根据经验，写下基于CPU。 本人电脑：windows10 操作系统+集显（基于CPU操作） 在这之前，还需要介绍几款英文名字之间联系： python：是一种高级编程语言，和C语言的性质一样，是编程的一种，类似于我们以前学习...
在tensorflow中设置使用某一块GPU、多GPU、CPU的操作
12-20
它支持在不同硬件上运行，包括CPU和GPU。在多GPU环境下，能够有效地分配计算任务可以显著提高训练速度。以下将详细解释如何在TensorFlow中设置使用特定GPU、多GPU以及仅使用CPU。 首先，如果你想要指定使用某一块...
conda安装CPU版本tensorflow
最新发布
03-11
嗯，用户想安装CPU版本的TensorFlow，使用conda。我之前处理过类似的问题，可能需要注意一些步骤和常见错误。首先，我应该回忆一下conda的基本命令，以及如何指定CPU版本的TensorFlow。 首先，conda安装包通常是...
TensorFlow 强制使用CPU
weixin_44493841的博客
12-24 1646
TensorFlow 强制使用CPU 因为显存只有8G,训练时各种报错。有时训练几次就崩了。所以直接使用cpu,有条件的还是使用gpu的快。 import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 强制使用cpu
限制TensorFlow只在CPU上运行的方法
耶律大石的博客
05-04 1623
限制TensorFlow只在CPU上运行的方法 import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 注：上述代码一定要放在import tensorflow或keras等之前，否则不起作用。 ...
tensorflow如何指定gpu运行还是cpu运行
wangmengmeng99的博客
05-20 1095
如果需要切换成CPU运算，可以调用tf.device(device_name)函数，其中device_name格式如/cpu:0其中的0表示设备号，TF不区分CPU的设备号，设置为0即可。在TensorFlow中，如果你安装的TensorFlow版本是GPU版本的，并且你的电脑配置符合gpu运行的条件，那么系统就默认使用gpu进行训练，不用单独去指明用GPU。run() 函数的运作计制是：先加载flags的参数项，然后执行main() 函数，其中参数使用tf.app.flags.FLAGS定义。
windows环境下Tensorflow环境搭建（CPU版）
Yuanbaokk的博客
07-12 4654
windows中Tensorflow环境搭建及PyCharm中配置tensorflow环境
tensorflow配置只使用CPU
dddeee的专栏
08-03 4812
tensorflow只使用cpu配置
tensorflow在CPU可以正常运行，在GPU报错
weixin_45885232的博客
02-06 794
tensorflow在GPU下报奇奇怪怪错误的解决方案，尤其是7209行错误
【TensorFlow】GPU/CPU硬件设置
莫邪莫急的博客
05-07 913
强制使用CPU 有的时候为了对比CPU和GPU运行效率，需要讲GPU版本的代码在CPU环境运行，但是因为存在GPU的缘故使得代码始终处于GPU环境运行。这个时候需要强制使用CPU跑程序： os.environ["CUDA_VISIBLE_DEVICES"] = "" 限制显存使用 TensorFlow调用GPU时默认采用贪婪模式，即无论模型大小，都将占用所有可用的显存。如果想要多线程运行同一个模...
Tensorflow中使用CPU及GPU的参数设置
akaigyouhou的博客
03-05 3521
Tensorflow中使用CPU及GPU的参数设置 在进行SVM机器学习训练时，报dst Tensorflow is not initialized错误，为显存使用溢出错误。解决方式为使用CPU或者限制GPU显存。 1、设置使用GPU还是CPU： import os os.environ['CUDA_VISIBLE_DEVICES'] = "0" 2、安装GPU版本Tensorflow，只有一...
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
表情包 代码片
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值