diff --git a/setup.py b/setup.py
index 2363ee4d6f9d0ad0ae459eda4331d6770c2c240a..66516fc727c977e6a8e1058dd960d694412a1cff 100755
--- a/setup.py
+++ b/setup.py
@@ -181,7 +181,7 @@ if BUILD_MASK & DS_BUILD_CPU_ADAM:
                               '-O3',
                               '--use_fast_math',
                               '-gencode',
-                              'arch=compute_61,code=compute_61',
+5                              'arch=compute_61,code=compute_61',
                               '-gencode',
                               'arch=compute_70,code=compute_70',
                               '-std=c++14',
@@ -216,6 +216,7 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
                               '--use_fast_math',
                               '-gencode',
                               'arch=compute_61,code=compute_61',
+                              'arch=compute_60,code=compute_60',
                               '-gencode',
                               'arch=compute_70,code=compute_70',
                               '-std=c++14',
@@ -247,6 +248,7 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
                               '--use_fast_math',
                               '-gencode',
                               'arch=compute_61,code=compute_61',
+                              'arch=compute_60,code=compute_60',
                               '-gencode',
                               'arch=compute_70,code=compute_70',
                               '-std=c++14',