禁用GPU版本TensorFlow，切换到CPU版本TensorFlow。

#禁用gpu版本TensorFlow，因为CUDA号码从0开始，这里直接让CUDA使用-1的GPU，自然就无法使用gpu了。

代码前面加入：

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf
Environment Variable Syntax   Results
CUDA_VISIBLE_DEVICES=1   Only device 1 will be seen
CUDA_VISIBLE_DEVICES=0,1   Devices 0 and 1 will be visible
CUDA_VISIBLE_DEVICES="0,1"   Same as above, quotation marks are optional
CUDA_VISIBLE_DEVICES=0,2,3   Devices 0, 2, 3 will be visible; device 1 is masked

如果安装了GPU，默认使用GPU版本的TensorFlow
---------------------
作者：湾区人工智能
来源：CSDN
原文：https://blog.csdn.net/BTUJACK/article/details/82622966
版权声明：本文为博主原创文章，转载请附上博文链接！

禁用GPU版本TensorFlow，切换到CPU版本TensorFlow。相关推荐

pytorch cpu版本安装_pytorch深度学习框架--gpu和cpu的选择
最近实现了一个简单的手写数字识别的程序,我安装的pytorch是gpu版(你也可以安装cpu版本的,根据个人需要),这里我介绍pytorch的gpu版本和cpu版本的安装以及训练手写数字识别时gpu和 ...
anaconda哪个版本是 python3.6_windows10（64位）Anaconda3+Python3.6搭建Tensorflow（cpu版本）及keras...
1.本来电脑安装的是anaconda3 5.3.1,但安装的python版本是3.7,后来卸载了安装anaconda3 4.4.0,这个版本是3.6 2.打开Anaconda Prompt conda ...
conda安装GPU版pytorch，结果却是cpu版本[找到问题根源，从容解决]
conda安装GPU版pytorch,结果却是cpu版本[找到问题根源,从容解决] 一.问题描述二.网上解决方案罗列[此节为反面方案罗列!!!] 三.发现的根本原因[独家] 3.1 pytorch文 ...
PaddleOCR——Docker环境下基于HubServing模式部署Restful API服务（CPU版本）
Docker环境下基于HubServing模式部署Restful API服务(CPU版本) 在日常项目应用中,相信大家一般都会希望能通过Docker技术,把PaddleOCR服务打包成一个镜像,以便在 ...
【使用 node 版本切换工具 nvm 切换 node.js 版本】
使用 node 版本切换工具 nvm 切换 node.js 版本文章目录使用 node 版本切换工具 nvm 切换 node.js 版本前言一.nvm是什么? 二.使用步骤 1.卸载原有 no ...
当涉及某个项目需要大量使用到tensorflow时，最后不要使用tensorflow的GPU版本，这会出很多毛病，最好使用CPU版本
当涉及某个项目需要大量使用到tensorflow时,最后不要使用tensorflow的GPU版本,这会出很多毛病,最好使用CPU版本
深度学习环境配置（GPU、CPU版本）
一.Anaconda安装 1.介绍 Anaconda是一个开源的Python发行版本,用来管理Python相关的包,安装Anaconda可以很方便的切换不同的环境,使用不同的深度学习框架开发项目,本文 ...
tensorfow2.0以上版本，ipynb文件一键安装！安装不成功我倒立洗头~ [gpu+cpu版本]
目录准备说明 1版本对应关系 2.我的系统环境 python环境准备更新显卡驱动 1.驱动版本要求对照表 2.查看自己的显卡驱动版本 3.下载更新安装cuda 安装cudnn 环境检查+tens ...
Ubuntu 16.04下Caffe-SSD的应用（一）——编译Caffe-SSD的CPU版本与GPU版本
前言 1.之前用过Caffe的GooglNet做图像分类的相关项目,比起传统的办法,效果要好很多,在目标检测这块,之前也用传统的HOG加SVM实现过车辆,行人和马匹的检测,但正样本要上万张图像,才达到 ...

禁用GPU版本TensorFlow，切换到CPU版本TensorFlow。

禁用GPU版本TensorFlow，切换到CPU版本TensorFlow。相关推荐

最新文章

热门文章