Ubuntu Pytorch/CUDA-找不到GPU

我正在尝试使用我的GPU来训练网络。我用anaconda安装了PyTorch和CUDA。我有一个NVIDA GTX 860M在我的电脑上，我想使用

使用命令时：

$ lspci | grep NVIDIA

我得到输出：

01:00.0 3D controller: NVIDIA Corporation GM107M [GeForce GTX 860M] (rev a2)

所以图形卡就在那里。但在Jupyter笔记本中，以下几行：

print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.version.cuda)

返回：

0
False
9.0.176

我曾尝试使用conda更新最新的Cuda驱动程序，但没有解决问题。我的PyTorch版本是0.4.1.post2。在浏览不同的在线论坛时，我找不到答案。 我在Ubuntu 18.04.1 LTS上。

---

提前谢谢

您是否尝试过使用Pytork的更新版本？可能是因为你的Pytork无法与CUDA版本相匹配。或者，您的CUDA版本可能与GPU驱动程序版本不匹配。您可以使用命令

nvidia smi

查看驱动程序版本。