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

时间:2019-07-24 03:47:03

标签: python tensorflow gpu nvidia

我正在使用

  

将tensorflow导入为tf

     
    

2.2.4-tf gpu
    泰坦xp-Geforce Nvidia
    RGB电缆屏幕

  

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

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

我正试图限制GPU的使用,因为
当我使用GPU进行游戏或深度学习时
屏幕开始闪烁
如果您还有其他解决方案,请告诉我

0 个答案:

没有答案