GPU版TensorFlow怎么指定让CPU运行

由于某些原因GPU版的TensorFlow运行起来会出现一些问题，比如内存溢出等情况。此时我们可以用CPU和系统内存来运行我们的程序。

代码如下：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
将以上三行代码放入程序首部即可。

GPU版TensorFlow怎么指定让CPU运行相关推荐

Ubuntu 16.04 LTS下编译GPU版tensorflow
Ubuntu 16.04 LTS下编译GPU版tensorflow 机器学习与数学 · 2016-06-10 13:51 作者: 比特小组机器学习与数学出品机器学习必然涉及到代码,本小组选择sci ...
Win10安装GPU版tensorflow和keras
Win10安装GPU版tensorflow和keras 1.python安装在此选用python3.7安装见详细教程https://blog.csdn.net/weixin_43545253/art ...
GTX1060安装gpu版tensorflow经验分享
配适版本作为一个深度学习小白,刚刚开始接触深度学习,没想到安装tensorflow竟然成为了第一支拦路虎-用CPU的版本完成作业后想着既然显卡还凑活不如用GPU版本的试试,没想到安装实在麻烦,网上的 ...
conda安装GPU版pytorch，结果却是cpu版本[找到问题根源，从容解决]
conda安装GPU版pytorch,结果却是cpu版本[找到问题根源,从容解决] 一.问题描述二.网上解决方案罗列[此节为反面方案罗列!!!] 三.发现的根本原因[独家] 3.1 pytorch文 ...
(转）在Windows上安装GPU版Tensorflow
转载自在Windows上安装GPU版Tensorflow. 1. 下载安装Anaconda 简单说就是下载 64位 python 3.5 版本的Anaconda https://www.continu ...
错误调试：GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
如果你是使用 GPU 版 TensorFlow 的话,并且你想在显卡高占用率的情况下(比如玩游戏)训练模型,那你要注意在初始化 Session 的时候为其分配固定数量的显存,否则可能会在开始训练的时候 ...
conda安装gpu版tensorflow和局部cuda插件，不用配置全局cuda了
conda安装gpu版tensorflow和局部cuda插件,不用配置全局cuda了把后面的命令改为你要安装的版本号 conda install -c conda-forge tensorflow- ...
ubuntu16.04下安装GTX1080TI显卡驱动+安装CUDA+Cudnn+anaconda+GPU版tensorflow
参考博客:https://blog.csdn.net/weixin_40294256/article/details/79157838 先安装好GTX1080Ti 显卡驱动--安装CUDA--安装Cu ...
tensorflow实践笔记：屏蔽GPU版tensorflow的提示、分布式训练部署时的问题以及tensorboard的查看
CPU版本的tensorflow代码改到GPU版运行时,会出现很多提示: 上面的提示可以通过加入下面的设置来屏蔽: import os os.environ['TF_CPP_MIN_LOG_LEVEL ...

GPU版TensorFlow怎么指定让CPU运行

GPU版TensorFlow怎么指定让CPU运行相关推荐

最新文章

热门文章