diff --git a/setup.py b/setup.py
index 2363ee4d6f9d0ad0ae459eda4331d6770c2c240a..1364caab23f9e085b4234e2f094d608866a9c7bc 100755
--- a/setup.py
+++ b/setup.py
@@ -217,6 +217,8 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
                               '-gencode',
                               'arch=compute_61,code=compute_61',
                               '-gencode',
+                              'arch=compute_60,code=compute_60',
+                              '-gencode',
                               'arch=compute_70,code=compute_70',
                               '-std=c++14',
                               '-U__CUDA_NO_HALF_OPERATORS__',
@@ -248,6 +250,8 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
                               '-gencode',
                               'arch=compute_61,code=compute_61',
                               '-gencode',
+                              'arch=compute_60,code=compute_60',
+                              '-gencode',
                               'arch=compute_70,code=compute_70',
                               '-std=c++14',
                               '-U__CUDA_NO_HALF_OPERATORS__',