diff --git a/tao_compiler/ci_build/platforms/tao/gpu/env.conf.cuda11_9 b/tao_compiler/ci_build/platforms/tao/gpu/env.conf.cuda11_9 new file mode 100644 index 00000000000..4280acca0f7 --- /dev/null +++ b/tao_compiler/ci_build/platforms/tao/gpu/env.conf.cuda11_9 @@ -0,0 +1,8 @@ +TF_NEED_CUDA=1 +TF_CUDA_CLANG=0 +TF_CUDA_VERSION=12.8 +TF_CUDNN_VERSION=8 +TF_CUDA_COMPUTE_CAPABILITIES="6.0,6.1,7.0,7.5,8.0,8.6,9.0" +TF_NEED_TENSORRT=0 +TF_NEED_ROCM=0 +TF_SET_ANDROID_WORKSPACE=0