diff --git a/ci/build_windows.py b/ci/build_windows.py index e0264675e8f2..7412a733bc90 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="5.2" ' + '-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="5.2" ' + '-DMXNET_CUDA_ARCH="7.5" ' '-DUSE_MKLDNN=ON ' '-DCMAKE_CXX_FLAGS="/FS /MD /O2 /Ob2" ' '-DCMAKE_BUILD_TYPE=Release')