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

时间:2016-09-19 22:52:27

标签: windows gpu nvidia caffe tesla

当我为Caffe win7使用两个不同的GPU时,它不起作用。 操作系统:win7 SP1 GPU:Quadro K620,Tesla K40 Caffe:Caffe for windows 问题是每次我更新到K620的最新驱动程序369.26,然后 特斯拉K40不起作用。然后当我更新到特斯拉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 个答案:

答案 0 :(得分:2)

我认为问题是每次在Win7上,nvidia update会将两个驱动程序集成到一个。然后,如果一个驱动程序更高一个,caffe要求两个驱动程序更新。因此,它不会起作用。