Драйвер NVIDIA не распознается при использовании Pytorch и Matplotlib

Question

Я пытался использовать Pytorch 0.4.0 в Python 3.5. У меня есть графический процессор NVIDIA K40c и драйвер CUDA версии V9.0.176. Однако, когда я выполняю «torch.cuda.is_available ()», он возвращает False. Итак, когда я пытался определить Тензор, он выдает следующую ошибку:

AssertionError:
Found no NVIDIA driver on your system. Please check that you
have an NVIDIA GPU and installed a driver from

Я сократил код до следующих трех строк:

import torch
import matplotlib.pyplot as plt
print("Is CUDA available? ",torch.cuda.is_available())

И результат все еще ложный. Однако, если я прокомментирую строку:

#import matplotlib.pyplot as plt

работает нормально. Похоже, что есть проблемы с совместимостью с matplotlib, есть идеи?

Answer 1

Я только что понял, похоже, что matplotlib маскирует некоторые функции Pytorch, поэтому, если я изменю порядок строк импорта, все будет отлично:

import matplotlib.pyplot as plt
import torch
print("Is CUDA available? ",torch.cuda.is_available())

Итак, теперь он возвращает True.