gpu選擇使用問題

2021-09-09 07:34:05 字數 1790 閱讀 1450

2023年10月31日 11:41:04

lscelory

伺服器中有多個gpu,選擇特定的gpu執行程式可在程式執行命令前使用:cuda_visible_devices=0命令。0為伺服器中的gpu編號,可以為0, 1, 2, 3等,表明對程式可見的gpu編號。

cuda_visible_devices=

1# 只有編號為1的gpu對程式是可見的,在**中gpu[0]指的就是這塊兒gpu

cuda_visible_devices=0,

2,3# 只有編號為0,2,3的gpu對程式是可見的,在**中gpu[0]指的是第0塊兒,

# gpu[1]指的是第2塊兒,gpu[2]指的是第3塊兒

cuda_visible_devices=2,

0,3# 只有編號為0,2,3的gpu對程式是可見的,但是在**中gpu[0]指的是第2塊兒,

# gpu[1]指的是第0塊兒,gpu[2]指的是第3塊兒--

----

----

----

----

---# 原文:

臨時設定:

linux: export cuda_visible_devices=

1windows:

set cuda_visible_devices=

1

永久設定:

linux:在~/

.bashrc 的最後加上export cuda_visible_devices=

1,然後source ~

/.bashrc

windows:

開啟我的電腦環境變數設定的地方,直接新增就行了。

伺服器中有多個gpu,選擇特定的gpu執行程式可在程式執行命令前使用:cuda_visible_devices=0命令。0為伺服器中的gpu編號,可以為0, 1, 2, 3等,表明對程式可見的gpu編號。

cuda_visible_devices=

1# 只有編號為1的gpu對程式是可見的,在**中gpu[0]指的就是這塊兒gpu

cuda_visible_devices=0,

2,3# 只有編號為0,2,3的gpu對程式是可見的,在**中gpu[0]指的是第0塊兒,

# gpu[1]指的是第2塊兒,gpu[2]指的是第3塊兒

cuda_visible_devices=2,

0,3# 只有編號為0,2,3的gpu對程式是可見的,但是在**中gpu[0]指的是第2塊兒,

# gpu[1]指的是第0塊兒,gpu[2]指的是第3塊兒--

----

----

----

----

---# 原文:

臨時設定:

linux: export cuda_visible_devices=

1windows:

set cuda_visible_devices=

1

永久設定:

linux:在~/

.bashrc 的最後加上export cuda_visible_devices=

1,然後source ~

/.bashrc

windows:

開啟我的電腦環境變數設定的地方,直接新增就行了。

強制使用GPU

對於tensorflow,宣告session的時候加入device count 即可 import tensorflow as tf sess tf.session config tf.configproto device count 對於keras,則呼叫後端函式,設定其使用如上定義的sessio...

gpu使用細節

在tensowflow 配置每個 gpu 上占用的記憶體的比例 gpu options tf.gpuoptions per process gpu memory fraction 0.95 sess tf.session config tf.configproto gpu options gpu o...

caffe matlab無法使用gpu

常見問題 1.版本不相容。我用的是vs2013 cuda7.5 cudnn的v5版 matlab2016a,如果選用的是microsoft的caffe,是可以用cudnn的v5版的。但cuda8.0尚不被支援,如果你的顯示卡是最新的gtx10系列,如果cuda7.5不能驅動的話,就只能再等等了。所以...