使用Pytorch和Matplotlib时无法识别NVIDIA驱动程序

Question

我一直在尝试在Python 3.5中使用Pytorch 0.4.0。我有一个NVIDIA K40c GPU和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())

结果仍然为False。但是，如果我注释以下行：

#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。