Tesla V100 GPU слишком медленный

431
Ant

Я использую Ubuntu 16.04 с графическим процессором Tesla V100. Я установил CUDA 9.1 и cuDNN 7. Версия драйвера 387.26. GPU правильно распознан; это появляется с nvidia-smi.

Я использую его с Pytorch 0,3. Проблема в том, что тренировка медленная; Одна эпоха обучения на cifar100 занимает около часа, что составляет лишь часть скорости, на которую способен V100. Я понятия не имею, в чем может быть проблема.

Есть ли у вас предложения?

0
взгляните на использование CUDA, а также расскажите, как именно вы получили дополнения Python: бинарные пакеты или скомпилированные из исходных текстов ** на вашем хосте **? Alexey Vesnin 5 лет назад 0

0 ответов на вопрос