Update start.sh

start.sh

@@ -1,6 +1,23 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
+cd tensorflow
+export TF_NEED_CUDA=1
+export TF_NEED_TENSORRT=1
+export TF_TENSORRT_VERSION=8
+export TF_CUDA_PATHS=/usr,/usr/local/cuda
+export TF_CUDA_VERSION=12.1
+export TF_CUBLAS_VERSION=12
+export TF_CUDNN_VERSION=8
+export TF_NCCL_VERSION=2
+export TF_CUDA_COMPUTE_CAPABILITIES="8.0,9.0"
+export TF_ENABLE_XLA=1
+export TF_NEED_HDFS=0
+export CC_OPT_FLAGS="-march=sandybridge -mtune=broadwell"
+yes "" | ./configure
+
+
+
 tree -L 4 /app
 tree -L 4 /data