1. 程式人生 > >GPU版TensorFlow怎麼指定讓CPU執行

GPU版TensorFlow怎麼指定讓CPU執行

由於某些原因GPU版的TensorFlow執行起來會出現一些問題,比如記憶體溢位等情況。此時我們可以用CPU和系統記憶體來執行我們的程式。

程式碼如下:

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

將以上三行程式碼放入程式首部即可。

相關推薦

GPUTensorFlow怎麼指定CPU執行

由於某些原因GPU版的TensorFlow執行起來會出現一些問題,比如記憶體溢位等情況。此時我們可以用CPU和系統記憶體來執行我們的程式。 程式碼如下: import os os.environ["

GPU tensorflow在訓練CIFAR資料集時GPU 沒有參與計算/出現假宕機狀態/CPU滿負荷運轉

程式碼:《TensorFlow實戰》黃文堅著,第5.3節 TensorFlow 實現進階的卷積網路,所用資料集為CIFAR-10 出現問題的電腦型號:thinkpad T470; 顯示卡: GeForce 940mx 問題描述:書中提供的程式碼之前在不同的桌上型電腦上執行

Windows64位安裝GPUTensorFlow 0.12,Power Shell下輸入:安裝Tensorflow的全教程

unless 設置環境變量 log api err 化工 查看 aid nbsp 推薦使用powershell,只需要在cmd指令窗口輸入powershell即可 下載64位Python3.5(一定要3.5!!)可以通過Python 3.5 from python.org

win7/win10下cuda8.0安裝跑gputensorflow

簡單寫下配置過程吧,挺簡單的: 0.win7/10 X64系統 1.安裝vs2013 2.安裝Anaconda3(需要裡面的python環境) 3.下載CUDA8.0(https://developer.nvidia.com/cuda-downloads),下面兩個exe檔案都下載 &

Fedora27搭建GPUTensorFlow

主機:華碩(ASUS)G11飛行堡壘 顯示卡:GeForce 1070 原作業系統為win10,準備安裝Fedora 27,從零開始搭建GPU版TensorFlow。 安裝Fedora27 下載ISO映象並製作U盤啟動盤(注意!啟動盤一定用官

Windows 10 安裝 Anaconda3 & GPUTensorflow & Cuda9 & cudnn7

前言 為了更好的學習工作,清理了下電腦,重做了下系統,因此需要重新裝深度學習相關的執行環境。在這裡記錄一下安裝過程,以備後用。 環境:Windows 10 & vs-2015 & cuda9 & cudnn7 & anaconda

Ubuntu16.04安裝配置GPUTensorFlow

基本工作 更新系統 sudo apt-get update sudo apt-get upgrade -y sudo apt-get dist-upgrade -y 安裝linux kernel modules,linux source和headers

ubantu安裝cuda和GPUtensorflow(詳細)

首先十分感謝我的老師對我的支援,終於申請到了電腦,現在終於可以在linux系統下載配置tensorflow-GPU加速器了,對於新手說安裝過程也是十分’感動’,掉坑無數,希望本文可以給大家帶來幫助。 步驟一:安裝Nvidia顯示卡驅動 步驟二:下載並安裝cu

Ubuntu16.04安裝CUDA+cuDNN+GPUTensorFlow過程記錄

一、安裝環境說明 軟體 版本 下載連結 說明 Ubuntu系統 16.04-64位 … … Python anaconda 3.6 不使用系統自帶的2.7版本 CUDA cuda_8

【深度學習】CentOS 7 安裝GPUTensorflow教程(一)

之前一直在玩cpu版的tensorflow,這些天突然心血來潮,想搞個gpu版的tensorflow來嚐嚐鮮,沒想到把所有能夠踩的坑幾乎全部踩了一遍,在這裡把自己踩的坑和一些安裝細節拿出來分享給大家,

Windows10 GPUTensorflow配置教程+Anaconda3+Jupyter Notebook

之前配Caffe費了不少周折,詳情參閱 深度學習之caffe入門——caffe環境的配置(CPU ONLY)。 如今轉戰Tensorflow,又免不了配環境之苦,摸索半天。終得其法。記錄下來,以備後用。 一、在使用pip安裝package的時候,經常崩掉,換

TensorFlow指定GPUCPU執行相應運算w

一、環境 TensorFlow API r1.12 CUDA 9.2 V9.2.148 Python 3.6.3 二、官方說明 https://www.tensorflow.org/api_docs/python/tf/device 使用預設圖(default graph)是

tensorflow指定gpucpu

如果機器中有多塊GPU,tensorflow會預設吃掉所有能用的視訊記憶體, 如果實驗室多人公用一臺伺服器,希望指定使用特定某塊GPU。 可以在檔案開頭加入如下程式碼: import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BU

TensorFlow】Windows環境下PyCharm執行TensorFlow GPU(附TensorFlow更新方法)

作為一名穀粉再加上最近研究深度學習,不得不說TensorFlow是一個很好的切入點,今天看了下官方教程踩了幾個坑後,終於在PyCharm上起飛了,廢話不多說下面開始搭建環境吧。(此教程Win10,8.1和7都適用,只需下載不同的安裝檔案即可) 1. 確定安

關於tensorflow gpucpu網上說的容易混淆的一點

1. 現在網上說的怎麼選tensorflow gpu版本或cpu版本訓練時,來來去去就是那些命令,如with tf.device等等。這時讀者其實有些只安裝tensorflow-gpu,有些是tensorflow-gpu和tensorflow(cpu版)一起裝了,因此會有不

如何程序執行指定cpu

1、在linux下修改程序的“cpu親和力”   taskset是LINUX提供的一個命令(ubuntu系統可能需要自行安裝,schedutils package)。他可以讓某個程式執行在某個(或)某些CPU上。 以下均以redis-server舉例。 1)顯示程序執行的C

TensorFlow——tensorflow指定CPUGPU運算

1.指定GPU運算 如果安裝的是GPU版本,在執行的過程中TensorFlow能夠自動檢測。如果檢測到GPU,TensorFlow會盡可能的利用找到的第一個GPU來執行操作。 如果機器上有超過一個可用的GPU,除了第一個之外的其他的GPU預設是不參與計算的。為了讓TensorFlow使用這些GPU,必須將

win7配置安裝tensorflow(非GPU

win7配置安裝tensorflow(非GPU版) 引言 1、配置安裝Anaconda3-4.2.0 2、通過Anaconda配置相應版本Python 3、配置安裝TensorFlow 4、下載python IDE pychar

限制TensorFlow只在CPU執行的方法

我們知道安裝tensorflow可以安裝兩個版本的,有cpu和gpu的,如果同時安裝的時候會預設進行用gpu進行,那如歌更改為使用cpu呢 在import tensorflow/keras 之前,加入如下程式碼 import os os.environ["CUDA_DEVICE_ORDER"] =

C++ 將程序執行指定CPU

方法:  SetProcessAffinityMask(HANDLE,DWORD); 其中,第一個引數為程序控制代碼。 如果要知道當前執行緒的控制代碼,可以通過函式:GetCurrentThread()得到。否則,在建立多執行緒的時候,也同樣可以得