Estou usando o Ubuntu 16.04 com uma GPU tesla V100. Eu instalei CUDA 9.1 e cuDNN 7. A versão do driver é 387.26. A GPU é reconhecida corretamente; ele aparece com nvidia-smi.
Estou usando com o pytorch 0.3. O problema é que o treinamento é meio de desacelerar; uma época de treinamento em cifar100 leva cerca de uma hora, o que é apenas uma fração da velocidade da capacidade do V100. Eu não tenho idéia do que o problema poderia ser.
Você tem alguma sugestão?
Tags gpu cuda nvidia-graphics-card