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

Question

我在我的 linux 机器上安装了 anaconda3 和 cuda 9（准确地说是 aws），当我按照官方文档检查 caffe2 安装时，它显示了成功消息，但是当我使用以下脚本检查 GPU 时，python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'我看到了以下结果Logging before InitGoogleLogging() is written to STDERR E0729 20:58:01.995159 2016 common_gpu.cc:42] Found an unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES。我在做什么我做错了。

Answer 1

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

1. 尝试重新启动您的计算机（实际尝试不会有什么坏处）
2. 完全重新安装 NVIDIA 驱动程序 + CUDA + CuDNN（检查环境变量）
3. 按照官方文档完全重新安装 Caffe2。

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

• 官方教程：https ://caffe2.ai/docs/docker-setup.html
• 英伟达 Docker：https ://github.com/NVIDIA/nvidia-docker