-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrun_configure_gpu_1.13
62 lines (42 loc) · 2.13 KB
/
run_configure_gpu_1.13
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#!/usr/bin/expect
spawn ./configure
expect "Please specify the location of python*"
send "\n"
expect "Please input the desired Python library path to use*"
send "\n"
expect "Do you wish to build TensorFlow with XLA JIT support?"
send "y\n"
expect "Do you wish to build TensorFlow with OpenCL SYCL support?"
send "n\n"
expect "Do you wish to build TensorFlow with ROCm support?"
send "n\n"
expect "Do you wish to build TensorFlow with CUDA support?"
send "y\n"
expect "Please specify the CUDA SDK version you want to use.*"
send "10.2\n"
expect "Please specify the location where CUDA 10.2 toolkit is installed. Refer to README.md for more details.*"
send "\n"
expect "Please specify the cuDNN version you want to use.*"
send "\n"
expect "Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details.*"
send "/usr/lib/x86_64-linux-gnu\n"
expect "Do you wish to build TensorFlow with TensorRT support?"
send "n\n"
expect "Please specify the locally installed NCCL version you want to use.*"
send "\n"
# Для видеокарт Tesla P4, Tesla V100, Tesla T4, GeForce GTX 1070[Ti]/1080[Ti], GeForce RTX 2070[Ti]/2080[Ti], MX150, MX250 (https://en.wikipedia.org/wiki/CUDA)
expect "Please note that each additional compute capability significantly increases your build time and binary size.*"
send "6.1,7.0,7.5\n"
expect "Do you want to use clang as CUDA compiler?"
send "n\n"
expect "Please specify which gcc should be used by nvcc as the host compiler.*"
send "\n"
# MPI нужно в случае, когда у нас несколько GPU в машине и обучать мы хотим на всех сразу
expect "Do you wish to build TensorFlow with MPI support?"
send "n\n"
# По умолчанию в linux использует флаг -march=native, который автоматически определяет, какие инструкции поддерживает CPU
expect "Please specify optimization flags to use during compilation when bazel option \"--config=opt\" is specified*"
send "\n"
expect "Would you like to interactively configure ./WORKSPACE for Android builds?"
send "n\n"
interact