Skip to content

Commit 9b7574b

Browse files
authored
Karen/ci fix gpu (#71)
* update TF_CUDA_COMPUTE_CAPABILITIES to 7.5 * one more place to set for TF_CUDA_COMPUTE_CAPABILITIES to 7.5
1 parent 90b2895 commit 9b7574b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow/tools/ci_build/linux/gpu/run_mkl.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,9 @@ pip list
3939

4040
# Run configure.
4141
export TF_NEED_CUDA=1
42-
export TF_CUDA_COMPUTE_CAPABILITIES=6.0
42+
export TF_CUDA_COMPUTE_CAPABILITIES=7.5
4343
export PYTHON_BIN_PATH=`which python`
44-
yes "" | TF_NEED_CUDA=1 TF_CUDA_COMPUTE_CAPABILITIES=6.0 $PYTHON_BIN_PATH configure.py
44+
yes "" | TF_NEED_CUDA=1 TF_CUDA_COMPUTE_CAPABILITIES=7.5 $PYTHON_BIN_PATH configure.py
4545

4646
# Run bazel test command. Double test timeouts to avoid flakes.
4747
# Setting KMP_BLOCKTIME to 0 lets OpenMP threads to sleep right after parallel execution

0 commit comments

Comments
 (0)