TensorFlow进行GPU和CPU切换(强制使用CPU)

最新推荐文章于 2025-04-11 23:00:38 发布
最新推荐文章于 2025-04-11 23:00:38 发布
阅读量2.3w 收藏 54
点赞数 22
分类专栏： TensorFlow
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/thisiszdy/article/details/89242019
版权
本文介绍了一种在代码中切换GPU和CPU的方法，通过在代码开始处设置环境变量CUDA_VISIBLE_DEVICES为-1，可以实现在不使用GPU的情况下运行程序，反之则启用GPU加速。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

在代码的开头，所有 import 之前，加入下面两行代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # 这一行注释掉就是使用gpu，不注释就是使用cpu
keras在TensorFlow-GPU和CPU中来回切换，以及GPU显存不足的缓解方式（切换原因：GPU显存实在不足）
weixin_58420301的博客
03-22 4168
TensorFlow中GPU显存不足的问题与CPU切换为CPU的问题
tensorflow使用GPU训练
ftfy123的博客
03-04 1万+
1. 查看gpu的使用详情： （1）查看gpu使用情况 nvidia-smi.exe # windows上 nvidia-smi -l #linux服务器上 # 显示的结果中 Volatile GPU-Util：浮动的GPU利用率； （2）linux上查看进程占用gpu的情况： ##实用技巧： ##如果你在linux终端运行深度学习python脚本文件，运行中发现占用多个GPU和内存...
Tensorflow CPU GPU切换
qq_44425969的博客
10-16 3772
Tensorflow使用GPU、CPU
TensorFlow充分并行化使用CPU
最新发布
piaopiaolanghua的专栏
04-11 283
本文主要讲如何设置配置让TensorFlow充分使用CPU资料，提升训练效率。
tensorflow中gpu和cpu切换
ccxcuixia
04-26 1861
两者同时存在时使用gpu 要想使用cpu，在代码里添加如下命令： matrix1=tf.constant([[3.,3.]]) matrix2=tf.constant([[2.],[2.]]) with tf.Session() as ses: with tf.device("/cpu:0"): product=tf.matmul(matrix1,matrix2) 说明...
tensorflow设置GPU
DRACOYU的博客
01-17 6168
一、设置GPU使用 1.一劳永逸的方法，直接在~/.bashrc中设置环境变量 CUDA_VISIBLE_DEVICES       （1）vim ~/.bashrc       （2）在.bashrc.最下方添加 export  CUDA_VISIBLE_DEVICES='XX'(这个xx是GPU的序号，比如我这边服务器上有8个GTX1080序号是0~7，可以填入任意多个序号，序号间用逗号隔...
单机tensorflow cpu/gpu切换
qq_29153321的博客
04-02 1897
gpu环境安装参考https://blog.csdn.net/qq_29153321/article/details/87203888 单机tensorflow cpu/gpu切换 python3 model.py（默认gpu下运行） CUDA_VISIBLE_DEVICES="cpu" python3 model.py（切换到gpu下运行） ...
tensorflow-GPU CPU和GPU的切换
qq_22829061的博客
12-25 929
在所有代码前 import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 在这里插入代码片 这一行注释掉就是使用gpu，不注释就是使用cpu
关于Tensorflow使用CPU报错的解决方式
12-20
在使用TensorFlow进行深度学习开发时，我们可能会遇到这样的问题：在尝试运行操作时，系统报错提示"Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/...
Ubuntu-TensorFlow-GPU config
CS_GaoMing的博客
03-14 383
系统说明： CUDA9.0要求GCC版本是5.x或者6.x，其他版本不可以，需要自己进行配置，通过以下命令才对GCC版本进行修改。 特别提醒：重启大法好. # method 1: sudo reboot # method 2: sudo shutdown now GCC的版本核对问题. 版本安装： # 事先可以先查看版本的问题，ubuntu查看方式为:program_name --versio...
sklearn支持gpu_anaconda3下配置python-3.5+tensorflow-gpu-1.9.0人脸识别项目环境
weixin_40003046的博客
12-22 1036
之前为了配置tensorflow-gpu的环境又是装cuda,又是装cudnn，还有tensoflow-gpu等等，，因为当时也是第一次搭建这个环境，所以完全是按照别人的搭建方法来一步一步的弄得，，后来我在给室友安装环境的时候，发现cuda,cudnn什么的完全不用自己安装，，，全部交给 anaconda3 (好东西)就行了Anaconda3安装几乎最后所有的东西都是用这个完成的，，所以先去安装这...
我既下载了tensorflow2.13又下载了tensorflow2.6-gpu，需不需要卸载tensorflow2.13.0
03-08
TensorFlow的GPU版本和CPU版本通常会有冲突，尤其是当它们被安装在同一个环境中时。如果用户没有使用虚拟环境，可能会导致依赖冲突。例如，tensorflow-gpu 2.6和tensorflow 2.13可能会有不同的依赖库版本，比如CUDA...
tensorflow2 GPU和CPU切换
Leo_whj的博客
09-16 1846
当你安装上gpu版本的tensorflow2时，默认情况下使用GPU，当你想要切换回CPU运行程序时， 你需要的代码开头添加一句代码 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
禁用GPU版本TensorFlow，切换到CPU版本TensorFlow。
kyle1314608的博客
07-22 232
#禁用gpu版本TensorFlow，因为CUDA号码从0开始，这里直接让CUDA使用-1的GPU，自然就无法使用gpu了。 代码前面加入： import osos.environ["CUDA_VISIBLE_DEVICES"]="-1" import tensorflow as tfEnvironment Variable Syntax ResultsCUDA_VISI...
tensorflow2.0 GPU和CPU切换
ZHE
02-20 5089
首先若不加任何配置情况下，是默认使用gpu的， 加上下面这句代码就使用cpu了。 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
关于TensorFlow的GPU设置
Alan的博客
04-04 1万+
在使用GPU版的TensorFlow跑程序的时候，如果不特殊写代码注明，程序默认是占用所有主机上的GPU，但计算过程中只会用其中一块。也就是你看着所有GPU都被占用了，以为是在GPU并行计算，但实际上只有其中一块在运行；另外的所有显卡都闲着，但其显存都被占用了，所以别人也用不了。不过这种情况通过在程序之前加三行代码就可以解决： import os os.environ["CUDA_DEVIC...
tensorflow如何切换CPU和GPU
高颜值的杀生丸(此博客转载自我的博客园)
09-07 2905
import os if Bert_Use_GPU: os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1 else: os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU 　　
tensorflow：GPU使用
热门推荐
fireflychh的博客
06-28 1万+
1、tensorflow支持的设备及选择方式 一般系统中有多种计算设备，tensorflow使用CPU和GPU，这些设备通过如下名称标识： “/cpu:0”    "/gpu:0"    "/gpu:1" 如果一个操作tensorflow既有CPU版本又有GPU版本实现，则tensorflow将会优先选择GPU来执行操作，如matmul支持CPU和GPU内核，在一个系统上有cpu:0和gpu
TensorFlow在指定CPU/GPU上运行的方法
xq920831的博客
09-03 4156
转自：http://wiki.jikexueyuan.com/project/tensorflow-zh/how_tos/using_gpu.html 使用 GPUs 支持的设备 在一套标准的系统上通常有多个计算设备. TensorFlow 支持 CPU 和 GPU 这两种设备. 我们用指定字符串 strings 来标识这些设备. 比如: "/cpu:0": 机器...
评论 6
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值