Skip to content

Commit a2d2bc8

Browse files
JAicewizardfacebook-github-bot
authored andcommitted
Handle insufficient driver gracefully (facebookresearch#4271)
Summary: Gracefully handle insufficient drivers (ex. no driver available.) Resolves facebookresearch#4251 Pull Request resolved: facebookresearch#4271 Reviewed By: mnorris11 Differential Revision: D72351969 Pulled By: ramilbakhshyiev fbshipit-source-id: de2b6f741087c59665e7f9f171ee6096c7eea39b
1 parent 4af0182 commit a2d2bc8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

faiss/gpu/utils/DeviceUtils.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ void setCurrentDevice(int device) {
3030
int getNumDevices() {
3131
int numDev = -1;
3232
cudaError_t err = cudaGetDeviceCount(&numDev);
33-
if (cudaErrorNoDevice == err) {
33+
if (cudaErrorNoDevice == err || cudaErrorInsufficientDriver == err) {
3434
numDev = 0;
3535
} else {
3636
CUDA_VERIFY(err);

0 commit comments

Comments
 (0)