import os
os.environ['TF_CUDNN_USE_AUTOTUNE']='0'
os.environ['TF_FORCE_GPU_ALLOW_GROWTH']='true'
tensorflow 1.13以上版本环境变量
最新推荐文章于 2024-03-03 20:32:07 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
import os
os.environ['TF_CUDNN_USE_AUTOTUNE']='0'
os.environ['TF_FORCE_GPU_ALLOW_GROWTH']='true'