限制skflow中的GPU內存分配

Question

我正在與Tensorflow和skflow一起訓練與其他人共享的EC2實例的網點。對於我們所有人能夠同時工作，我想限制分配的可用GPU內存的一小部分。

This question它與Tensorflow，但因爲我使用sklfow我從來沒有使用tf.Session()。

可以通過skflow做同樣的事情嗎？

Answer 1

此時，只能通過將此參數傳遞給估計器來控制估計器中使用的內核數（num_cores）。

您可以將gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)添加到tf.ConfigProto，建議您鏈接到this question以實現您的需要。

隨意提交PR進行更改here以及向所有估算人員添加此附加參數。否則，我會在本週進行一些更改。

編輯：

我已經做了更改以允許這些選項。請查看示例文件夾中的「使用不同的GPU配置構建模型」示例。讓我知道是否有任何特殊需要或其他選項要添加。拉請求總是歡迎！