EuuIia commited on
Commit
cf67057
·
verified ·
1 Parent(s): e243096

Update start.sh

Browse files
Files changed (1) hide show
  1. start.sh +3 -3
start.sh CHANGED
@@ -1,7 +1,7 @@
1
  #!/usr/bin/env bash
2
  set -euo pipefail
3
 
4
- cd tensorflow
5
  export TF_NEED_CUDA=1
6
  export TF_NEED_TENSORRT=1
7
  export TF_TENSORRT_VERSION=8
@@ -10,11 +10,11 @@ export TF_CUDA_VERSION=12.1
10
  export TF_CUBLAS_VERSION=12
11
  export TF_CUDNN_VERSION=8
12
  export TF_NCCL_VERSION=2
13
- export TF_CUDA_COMPUTE_CAPABILITIES="8.0,9.0"
14
  export TF_ENABLE_XLA=1
15
  export TF_NEED_HDFS=0
16
  export CC_OPT_FLAGS="-march=sandybridge -mtune=broadwell"
17
- yes "" | ./configure
18
 
19
 
20
 
 
1
  #!/usr/bin/env bash
2
  set -euo pipefail
3
 
4
+
5
  export TF_NEED_CUDA=1
6
  export TF_NEED_TENSORRT=1
7
  export TF_TENSORRT_VERSION=8
 
10
  export TF_CUBLAS_VERSION=12
11
  export TF_CUDNN_VERSION=8
12
  export TF_NCCL_VERSION=2
13
+ #export TF_CUDA_COMPUTE_CAPABILITIES="8.0,9.0"
14
  export TF_ENABLE_XLA=1
15
  export TF_NEED_HDFS=0
16
  export CC_OPT_FLAGS="-march=sandybridge -mtune=broadwell"
17
+
18
 
19
 
20