Я пытаюсь сделать инструмент CUDA, и я должен сделать его масштабируемым. Я должен дать количество GPU, который система имеет как аргумент моему инструменту. Скажите мне, если существует какая-либо команда для получения количества GPU непосредственно??
Можно использовать sudo lshw -C display
перечислять Вашу видеокарту (видеокарты), затем Google каждый для наблюдения, сколько ядер это имеет.
Эта команда получает количество GPU непосредственно, предполагая, что Вы имеете nvidia-smi
.
nvidia-smi --query-gpu=name --format=csv,noheader | wc -l
Это печатает названия GPU, один на строку, и затем считает количество строк.