find-avaGPU.py

22 次浏览 2024-08-12 0 条评论

py

GPU Selection PyTorch Inference Memory Management

使用python，自动寻找服务器中显存最大的GPU进行pytorch推理，避免显存不够导致的推理失败。

文件大小：1.54KB

相关推荐