小型Python库自动将CUDA_VISIBLE_DEVICES设置为多GPU系统上最小负载的设备

小型Python库自动将CUDA_VISIBLE_DEVICES设置为多GPU系统上最小负载的设备

setGPU

A small Python library that automatically sets CUDA_VISIBLE_DEVICES to the least-loaded GPU on multi-GPU systems.

Installation: pip install setGPUUsage: import setGPU before any import that will use a GPU like torch or tensorflow.

Dependencies

Jongwook Choi's gpustat library.

Licensing

This code is in the public domain.

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。