diff --git a/lib/layer_utils/roi_pooling/build.py b/lib/layer_utils/roi_pooling/build.py
index 620867b..f634645 100644
--- a/lib/layer_utils/roi_pooling/build.py
+++ b/lib/layer_utils/roi_pooling/build.py
@@ -8,7 +8,10 @@
 defines = []
 with_cuda = False
 
-if torch.cuda.is_available():
+env_cuda = os.getenv('BUILD_WITH_CUDA', False)
+print(env_cuda)
+
+if torch.cuda.is_available() or env_cuda:
     print('Including CUDA code.')
     sources += ['src/roi_pooling_cuda.c']
     headers += ['src/roi_pooling_cuda.h']
diff --git a/lib/nms/build.py b/lib/nms/build.py
index 4d9a96b..8076a93 100644
--- a/lib/nms/build.py
+++ b/lib/nms/build.py
@@ -8,7 +8,10 @@
 defines = []
 with_cuda = False
 
-if torch.cuda.is_available():
+env_cuda = os.getenv('BUILD_WITH_CUDA', False)
+print(env_cuda)
+
+if torch.cuda.is_available() or env_cuda:
     print('Including CUDA code.')
     sources += ['src/nms_cuda.c']
     headers += ['src/nms_cuda.h']