diff --git a/ci/build_windows.py b/ci/build_windows.py index 74e9260a5a2a..91619e09b75a 100755 --- a/ci/build_windows.py +++ b/ci/build_windows.py @@ -141,7 +141,7 @@ class BuildFlavour(Enum): '-DUSE_BLAS=open ' '-DUSE_LAPACK=ON ' '-DUSE_DIST_KVSTORE=OFF ' - '-DMXNET_CUDA_ARCH="7.0" ' + '-DMXNET_CUDA_ARCH="7.5" ' '-DCMAKE_CXX_FLAGS="/FS /MD /O2 /Ob2" ' '-DUSE_MKL_IF_AVAILABLE=OFF ' '-DCMAKE_BUILD_TYPE=Release') @@ -159,7 +159,7 @@ class BuildFlavour(Enum): '-DUSE_BLAS=open ' '-DUSE_LAPACK=ON ' '-DUSE_DIST_KVSTORE=OFF ' - '-DMXNET_CUDA_ARCH="7.0" ' + '-DMXNET_CUDA_ARCH="7.5" ' '-DUSE_MKLDNN=ON ' '-DCMAKE_CXX_FLAGS="/FS /MD /O2 /Ob2" ' '-DCMAKE_BUILD_TYPE=Release') @@ -285,4 +285,4 @@ def main(): if __name__ == '__main__': - sys.exit(main()) \ No newline at end of file + sys.exit(main())