如果要在python代码中设置使用的GPU(如使用pycharm进行调试时)可以使用下面的代码(见第二个参考网址中Yaroslav Bulatov的回复):
默认tensorflow是使用GPU尽可能多的显存。可以通过下面的方式来设置使鼡的GPU显存:
可以按照需要,设置不同的值来分配显存。
上面的只能设置固定的大小如果想按需分配,可以使用allow_growth参数(参考网址:):
allow_growth