方法1:mxnet
import mxnet as mx
mx.test_utils.list_gpus()
# 或
mx.util.get_gpu_count()
方法2: nvgpu
pip install nvgpu
import nvgpu
nvgpu.available_gpus()
import mxnet as mx
mx.test_utils.list_gpus()
# 或
mx.util.get_gpu_count()
pip install nvgpu
import nvgpu
nvgpu.available_gpus()
本文标题:python获取GPU数量
本文链接:https://www.haomeiwen.com/subject/fzorvhtx.html
网友评论