diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh index f05fc6484290..5b1f9f6f21f7 100755 --- a/ci/docker/runtime_functions.sh +++ b/ci/docker/runtime_functions.sh @@ -347,7 +347,7 @@ build_centos7_gpu() { USE_CUDA_PATH=/usr/local/cuda \ USE_CUDNN=1 \ USE_DIST_KVSTORE=1 \ - CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ + # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ -j$(nproc) } @@ -643,7 +643,7 @@ build_ubuntu_gpu_mkldnn() { USE_CUDA=1 \ USE_CUDA_PATH=/usr/local/cuda \ USE_CUDNN=1 \ - CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ + # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ USE_SIGNAL_HANDLER=1 \ -j$(nproc) } @@ -660,7 +660,7 @@ build_ubuntu_gpu_mkldnn_nocudnn() { USE_CUDA=1 \ USE_CUDA_PATH=/usr/local/cuda \ USE_CUDNN=0 \ - CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ + # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ USE_SIGNAL_HANDLER=1 \ -j$(nproc) } @@ -679,7 +679,7 @@ build_ubuntu_gpu_cuda100_cudnn7() { USE_CUDNN=1 \ USE_CPP_PACKAGE=1 \ USE_DIST_KVSTORE=1 \ - CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ + # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \ USE_SIGNAL_HANDLER=1 \ -j$(nproc) }