diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..796b96d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/build
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2c570e0..a819ee0 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -44,11 +44,14 @@ if (WITH_OMP)
     set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Xcompiler -fopenmp")
 endif()
 
+# Remove deprecated compute archs only if not CUDA 11.0
+IF(NOT (CUDA_VERSION GREATER 10.2))
 # need to be at least 30 or __shfl_down in reduce wont compile
-set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_30,code=sm_30 -O2")
-set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_35,code=sm_35")
+  set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_30,code=sm_30 -O2")
+  set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_35,code=sm_35")
+  set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_50,code=sm_50")
+ENDIF()
 
-set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_50,code=sm_50")
 set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_52,code=sm_52")
 IF(CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 5)
   SET(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -D_MWAITXINTRIN_H_INCLUDED -D_FORCE_INLINES")
diff --git a/pytorch_binding/setup.py b/pytorch_binding/setup.py
index ad960ea..4506029 100644
--- a/pytorch_binding/setup.py
+++ b/pytorch_binding/setup.py
@@ -35,7 +35,7 @@
            "Build warp-rnnt and set WARP_RNNT_PATH to the location of"
            " libwarprnnt.so (default is '../build')").format(warp_rnnt_path))
     sys.exit(1)
-include_dirs = [os.path.realpath('../include')]
+include_dirs = [os.path.realpath('../include'), '/usr/local/cuda/include/']
 
 setup(
     name='warprnnt_pytorch',