tensorflow设置动态显存

2023-01-09

# 需要放在tensorflow调用前 , 动态显存,不要全部占用

os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"