From 9d4207616115b2a57d516d6f71df79619883c7dc Mon Sep 17 00:00:00 2001 From: Cresson Remi Date: Tue, 2 Jan 2024 21:14:00 +0100 Subject: [PATCH] COMP: update compute capability --- tools/docker/build-env-tf.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/docker/build-env-tf.sh b/tools/docker/build-env-tf.sh index 94546bf..e7703f0 100644 --- a/tools/docker/build-env-tf.sh +++ b/tools/docker/build-env-tf.sh @@ -47,7 +47,7 @@ if [ ! -z $CUDA_TOOLKIT_PATH ] ; then fi export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_TOOLKIT_PATH/lib64:$CUDA_TOOLKIT_PATH/lib64/stubs" export TF_CUDA_VERSION=$(echo $CUDA_TOOLKIT_PATH | sed -r 's/.*\/cuda-(.*)/\1/') - export TF_CUDA_COMPUTE_CAPABILITIES="5.2,6.1,7.0,7.5,8.6" + export TF_CUDA_COMPUTE_CAPABILITIES="5.2,6.1,7.0,7.5,8.0,8.6,9.0" export TF_NEED_CUDA=1 export TF_CUDA_CLANG=0 export TF_NEED_TENSORRT=0