@@ -11,40 +11,32 @@ export TF_NAME='tensorflow'
11
11
if [[ " $TF_VERSION " =~ ^2\. (16)\. [0-9]+$ ]] ; then
12
12
export BUILD_IMAGE=" tfra/nosla-cuda12.3-cudnn8.9-ubuntu20.04-manylinux2014-python$PY_VERSION "
13
13
export TF_CUDA_VERSION=" 12.3"
14
- export TF_CUDNN_VERSION=" 8.9"
15
14
export TF_USE_LEGACY_KERAS=1
16
15
elif [[ " $TF_VERSION " =~ ^2\. (15)\. [0-9]+$ ]] ; then
17
16
export BUILD_IMAGE=" tfra/nosla-cuda12.2-cudnn8.9-ubuntu20.04-manylinux2014-python$PY_VERSION "
18
17
export TF_CUDA_VERSION=" 12.2"
19
- export TF_CUDNN_VERSION=" 8.9"
20
18
elif [[ " $TF_VERSION " =~ ^2\. (14)\. [0-9]+$ ]] ; then
21
19
export BUILD_IMAGE=" tfra/nosla-cuda11.8-cudnn8.7-ubuntu20.04-manylinux2014-python$PY_VERSION "
22
20
export TF_CUDA_VERSION=" 11.8"
23
- export TF_CUDNN_VERSION=" 8.7"
24
21
elif [[ " $TF_VERSION " =~ ^2\. (12| 13)\. [0-9]+$ ]] ; then
25
22
export BUILD_IMAGE=" tfra/nosla-cuda11.8-cudnn8.6-ubuntu20.04-manylinux2014-python$PY_VERSION "
26
23
export TF_CUDA_VERSION=" 11.8"
27
- export TF_CUDNN_VERSION=" 8.6"
28
24
elif [[ " $TF_VERSION " =~ ^2\. ([6-9]| 10| 11)\. [0-9]+$ ]] ; then
29
25
export BUILD_IMAGE=" tfra/nosla-cuda11.2-cudnn8-ubuntu20.04-manylinux2014-python$PY_VERSION "
30
26
export TF_CUDA_VERSION=" 11.2"
31
- export TF_CUDNN_VERSION=" 8.1"
32
27
elif [ $TF_VERSION == " 2.4.1" ] ; then
33
28
export BUILD_IMAGE=' tfra/nosla-cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython'
34
29
export TF_CUDA_VERSION=" 11.0"
35
- export TF_CUDNN_VERSION=" 8.0"
36
30
elif [ $TF_VERSION == " 1.15.2" ] ; then
37
31
export BUILD_IMAGE=' tfra/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010-multipython'
38
32
export TF_CUDA_VERSION=" 10.0"
39
- export TF_CUDNN_VERSION=" 7.6"
40
33
else
41
34
echo " TF_VERSION is invalid: $TF_VERSION !"
42
35
exit 1
43
36
fi
44
37
45
38
echo " BUILD_IMAGE is $BUILD_IMAGE "
46
39
echo " TF_CUDA_VERSION is $TF_CUDA_VERSION "
47
- echo " TF_CUDNN_VERSION is $TF_CUDNN_VERSION "
48
40
49
41
if [ -z $HOROVOD_VERSION ] ; then
50
42
export HOROVOD_VERSION=' 0.28.1'
@@ -66,7 +58,6 @@ DOCKER_BUILDKIT=1 docker build --no-cache \
66
58
--build-arg TF_NAME \
67
59
--build-arg TF_NEED_CUDA \
68
60
--build-arg TF_CUDA_VERSION \
69
- --build-arg TF_CUDNN_VERSION \
70
61
--build-arg HOROVOD_VERSION \
71
62
--build-arg BUILD_IMAGE \
72
63
--build-arg NIGHTLY_FLAG \
0 commit comments