Amazon ec2 在anaconda中使用Caffe2获取GPU信息

我已经在我的linux机器上安装了anaconda3和cuda 9（准确地说是aws），当我检查caffe2安装时，正如在官方文档中一样，它会显示成功消息，但当我使用caffe2.python导入工作区中的以下脚本

python2-c'检查GPU时；print（workspace.NumCudaDevices（））”

在将InitGoogleLogging（）写入STDERR之前，我看到了以下结果

---

E0729 20:58:01.995159 2016 common_gpu.cc:42]发现未知错误-这可能是由于环境设置不正确，例如更改环境变量CUDA_VISIBLE_DEVICES。我做错了什么

您的问题转化为您的安装问题。 通常，通过以下措施之一解决此问题：

试着重新启动你的电脑（尝试一下不会有任何伤害）

完全重新安装NVIDIA驱动程序+CUDA+CuDNN（检查环境变量）

按照官方文档完全重新安装Caffe2

调试有故障的安装可能非常麻烦。如果您在Linux上运行，我建议您尝试Docker+Nvidia驱动程序：

• 官方教程：
• 英伟达码头工人：