如何限制2.2.4的gpu内存使用设置

Question

我在用

import tensorflow as tf

2.2.4-tf gpu
titan xp - geforce Nvidia
Rgb cable screen

要限制设备1的gpu内存使用量吗

os.environ["CUDA_VISIBLE_DEVICES"]="1"; 
# limit the memory usage of gpu 1

我试图限制gpu的使用，因为
当我使用gpu进行游戏或深度学习时
屏幕开始闪烁
如果你有别的解决办法，请告诉我