python中用于分配tensorflow gpu内存的包
configure-gpu的Python项目详细描述
GPU分配器
安装
$ pip install configure_gpu
这是一个用于分配GPU内存以防止堵塞GPU内存的包。
用法
# tensorflow importimporttensorflowastf# gpu allocator packagefromconfigure_gpuimportgpu_allocator# configuring gpugpu_memory_allocator(tf,percentage_allocation=0.95)