diff --git a/faiss/gpu/utils/DeviceUtils.cu b/faiss/gpu/utils/DeviceUtils.cu index 7a22c09e8f..39a7e71d76 100644 --- a/faiss/gpu/utils/DeviceUtils.cu +++ b/faiss/gpu/utils/DeviceUtils.cu @@ -30,7 +30,7 @@ void setCurrentDevice(int device) { int getNumDevices() { int numDev = -1; cudaError_t err = cudaGetDeviceCount(&numDev); - if (cudaErrorNoDevice == err) { + if (cudaErrorNoDevice == err || cudaErrorInsufficientDriver == err) { numDev = 0; } else { CUDA_VERIFY(err);