tensorflow中使用GPU的設定方法

2021-09-24 22:49:53 字數 796 閱讀 3004

方法一:

config=tf.configproto(allow_soft_placement=true)  ##:如果你指定的裝置不存在,允許tf自動分配裝置

config.gpu_options.allow_growth=true ##動態分配記憶體

sess=tf.session(config=config)

方法二:

gpu_options = tf.gpuoptions(allow_growth=true)

gpu_options =tf.gpuoptions(per_process_gpu_memory_fraction=0.8,allow_growth=true) ##每個gpu占用0.8 的視訊記憶體

config=tf.configproto(gpu_options=gpu_options,allow_soft_placement=true)

sess=tf.session(config=config)##如果電腦有多個gpu,tensorflow預設全部使用。如果想只使用部分gpu,可以設定cuda_visible_devices。

控制使用哪個gpu

os.environ[「cuda_device_order」] = 「pci_bus_id」 ##指定的裝置名稱

os.environ[『cuda_visible_devices』] = 『0』 #使用 gpu 0

os.environ[『cuda_visible_devices』] = 『0,1』 # 使用 gpu 0,1

Tensorflow中使用CPU及GPU的引數設定

在進行svm機器學習訓練時,報dst tensorflow is not initialized錯誤,為視訊記憶體使用溢位錯誤。解決方式為使用cpu或者限制gpu視訊記憶體。1 設定使用gpu還是cpu import os os.environ cuda visible devices 0 2 安裝...

GP服務的使用詳解

廢話不多說,開始正題 gp工具 它是由arcmap等桌面的gis應用生成的,生成的方式是使用模型構建器將需要的工具和引數拖進去,然後把工具和引數進行關聯,最後生成乙個流程圖。將流程圖進行驗證,如果成功了就可以進行發布,發布後就是gp工具了。gp工具的具體製作過程就不贅述了,網上有著相關的教程。gp工...

如何在C語言中使用tensorflow(一)概述

tensorflow是當前廣泛使用的機器學習框架。tensorflow提供了c語言的lib庫介面,但是對應的應用文件較少,這裡我將結合自己的使用情況做一下分享。先說一些我對機器學習實際應用的理解。一般來說機器學習功能的工程化應用分為線下訓練和線上部署兩個部分。線下訓練推薦採用python指令碼語言進...