使用GPU训练需要修改哪些参数

感谢您的工作
我已经在电脑上成功运行了代码，但是貌似只在CPU上工作，我已经设置了gpu-options-allow-growth=True，且能看到有一部显存占用，但似乎训练都是再CPU上完成的，我需要调整别的什么配置么?多谢