to solve the problem that a process occupies most of GPU memory
cella
2020
Branchable
0
12731
allow-growth - session config 에서 set하는 방법 - environment variable set 하는 라인을 main() 에 넣는 방법 os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true' # for the case "sessionConfiggpu_options.allow_growth=True" is not working