方法一：這個會使用最小的GPU資源

config = tf.ConfigProto() config.gpu_options.allow_growth = True self.sess=tf.Session(config=config)

方法二：

config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 占用GPU40%的顯存 session = tf.Session(config=config)

?

總結(jié)

以上是生活随笔為你收集整理的基础功能-tensorflow使用gpu的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。