tensorflow gpu記憶體使用問題

2021-08-14 17:55:47 字數 641 閱讀 8449

tensorflow對於gpu的使用是先佔住所有可用資源,然後使用某塊gpu進行計算,因此在**中需要對gpu的使用做一定的限定根據自己的需求。

當你不想讓tensorflow占用你所有的gpu時,可限定tensorflow的可見gpu

cuda_visible_devices=1 設定 1 為可見gpu 等同於 cuda_visible_devices=「1」

cuda_visible_devices=「0,1」設定 0 和 1為可見gpu

import os

os.environ["cuda_visible_devices"]="0"

config=tf.configproto()

config.gpu_options.per_process_gpu_memory_fraction=0.8 #占用gpu 80%的視訊記憶體

sess=tf.session(config=config)

config=tf.configproto()

config.gpu_options.allow_growth=true

sess=tf.session(config=config)

使用可見gpu的某塊gpu進行計算,gpu是從 0 開始的

with tf.device("/gpu:0")

離線安裝tensorflow gpu

前言 安裝前先確定電腦可以使用gpu執行tensorflow,否則浪費時間。nvidia 英偉達 官網官網檢視顯示卡是否支援gpu執行,compute capability需要在3.0及以上,否則安裝後,不能用gpu執行程式,會自動呼叫cpu執行。nvidia官網 密碼 0pmq 檔案列表 anac...

Tensorflow gpu版本安裝

這裡安裝想要嘗試最新的cuda10.1,但是失敗了,得到的教訓就是要嚴格的按照官方的版本要求。因為tensorflow1.13是基於cuda10的並沒有整合10.1,個人推測,兩家是不同的公司cuda10.1正常不會想到要相容自己的上層應用,或者s是tf直接寫死的需要cuda10。不管怎樣,現在總結...

TensorFlow GPU環境搭建

你們可能看不懂,因為主要是為了自己看 笑哭 顯示卡product gp104 geforce gtx 1080 10de 1b80 版本 ubantu 18.04 anaconda3 nvidia 930 cuda 91 cudnn 7.04 tensorflow gpu 1.13.1 一 uban...