From 5480ed5ea9d154375ae7481b4cae93b417524c92 Mon Sep 17 00:00:00 2001 From: Per Goncalves da Silva Date: Fri, 29 Mar 2019 14:36:44 +0100 Subject: [PATCH] Comments out CUDA_ARCH setting for compilation - Test --- ci/docker/runtime_functions.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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) }