Python 2.7 Caffe2不检测GPU

我想使用支持GPU的caffe2。我成功地用conda环境安装了caffe2（Ubuntu 16.04，python2.7）（命令：

conda每晚安装pytorch-c pytorch

）

它已成功安装（我使用caffe2.python import core'2>/dev/null&&echo“Success”| | echo“Failure”中的命令检查了它，并显示“Success”）

但是，当我检查caffe2 GPU构建（caffe2.python导入工作区中的命令：

python2-c；print（workspace.NumCudaDevices（））

）时，它返回0

---

我已经有了cuda、cuDNN、nccl，我不明白为什么caffe2无法检测可用的GPU。

我想你将要实现Detectron（否则这些天没人想使用这个愚蠢的caffe2）

我很确定这是由CUDA版本和CuDNN不匹配造成的。我被这个问题困扰了一段时间（你不知道哪个版本适合Caffe2），最后，我几乎同时得到了两个解决方案。他们俩都为我工作

首先，请将英伟达驱动程序更新到最新版本。我的版本更新为410.78，您只需在系统设置->软件和更新->其他驱动程序中选择特定驱动程序，即可更新驱动程序

别忘了重新启动你的电脑

然后，有两种方法来实现它

使用Docker构建环境

它简单快速。您只需安装Docker（以及用于GPU的nvidia Docker）并使用此预实现的环境和以下命令：

sudo docker pull ylashin/detectron

sudo nvidia docker run--rm-it ylashin/detectron

然后可以使用NumCudaDeivce命令测试Caffe2

这对我有用

感谢您的努力，请参见此处的全部介绍：

如果您在Docker安装方面有问题（特别是对于nvidia Docker），您可以直接跳到下一个

使用Detectron2

最新的Detectron最近出版（实际上是三天前！）。我们现在可以在Pytorch的支持下进行这项工作

以下是Detectron2：

---

只需转到最新的一个，您甚至可以在Google Colab中分发所有内容，这要简单得多。

您有哪些GPU？我使用NVIDIA GPU:Quadro M300M