Caffe on Windows common.cpp:166] 检查失败：错误 == cudaSuccess

Posted 2023-04-15

【中文标题】Caffe on Windows common.cpp:166] 检查失败：错误 == cudaSuccess

【英文标题】：Caffe on Windows common.cpp:166] Check failed: error == cudaSuccess

【发布时间】：2017-01-27 17:43:13

【问题描述】：

当我为 Caffe win7 使用两个不同的 GPU 时，它不起作用。 操作系统系统：win7 SP1 GPU：Quadro K620、Tesla K40 Caffe：适用于 Windows 的 Caffe 问题是每次我更新到 K620 的最新驱动程序 369.26 时，然后 特斯拉 K40 将无法工作。然后当我更新到 Tesla K40 的最新驱动程序 341.96 时， 那么K620将无法工作。因此，当我使用双 GPU 和错误时，它不适用于 caffe 是：

C:\Softwares\caffe-windows>.\Build\x64\Release\caffe.exe train --gpu=0,1 --solve
r=./examples/mnist/lenet_solver.prototxt
I0919 22:42:21.715857  5568 caffe.cpp:186] Using GPUs 0, 1
I0919 22:42:21.725857  5568 caffe.cpp:191] GPU 0: @■0
I0919 22:42:21.725857  5568 caffe.cpp:191] GPU 1: @■0
F0919 22:42:21.725857  5568 common.cpp:160] Check failed: error == cudaSuccess (
35 vs. 0)  CUDA driver version is insufficient for CUDA runtime version
*** Check failure stack trace: ***

如果我只使用 GPU Tesla K40（仅使用 Quadro，它可以工作），错误是：

C:\Softwares\caffe-windows>.\Build\x64\Release\caffe.exe train --gpu 1 --solver=
./examples/mnist/lenet_solver.prototxt
I0919 22:20:00.002020  4168 caffe.cpp:186] Using GPUs 1
I0919 22:20:00.095620  4168 caffe.cpp:191] GPU 1: @■/
F0919 22:20:00.095620  4168 common.cpp:166] Check failed: error == cudaSuccess (
10 vs. 0)  invalid device ordinal
*** Check failure stack trace: **

【参考方案1】：

我认为问题是每次在Win7上，nvidia更新都会将两个驱动程序合二为一。然后，如果一个驱动程序比一个驱动程序更高，并且 caffe 要求两个驱动程序是最新的。因此，它不会起作用。