Блог

Я пытался использовать Pytorch 0.4.0 в Python 3.5. У меня графический процессор NVIDIA K40c и драйвер CUDA версии 9.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())
  

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

 #import matplotlib.pyplot as plt
  

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

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

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

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