diff --git a/src/cnn/train.c b/src/cnn/train.c
index ba84848..4fcc00b 100644
--- a/src/cnn/train.c
+++ b/src/cnn/train.c
@@ -98,6 +98,7 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
     bool compatibility = check_cuda_compatibility();
     if (!compatibility) {
         printf("Exiting.\n");
+        exit(1);
     }
     #endif
     srand(time(NULL));