Skip to content

Commit 9afee0e

Browse files
Lifannoppenheimli
authored andcommitted
Remove cudnn from CI procedure
1 parent a21175a commit 9afee0e

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

.github/workflows/make_wheel_Linux_x86.sh

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,40 +11,32 @@ export TF_NAME='tensorflow'
1111
if [[ "$TF_VERSION" =~ ^2\.(16)\.[0-9]+$ ]] ; then
1212
export BUILD_IMAGE="tfra/nosla-cuda12.3-cudnn8.9-ubuntu20.04-manylinux2014-python$PY_VERSION"
1313
export TF_CUDA_VERSION="12.3"
14-
export TF_CUDNN_VERSION="8.9"
1514
export TF_USE_LEGACY_KERAS=1
1615
elif [[ "$TF_VERSION" =~ ^2\.(15)\.[0-9]+$ ]] ; then
1716
export BUILD_IMAGE="tfra/nosla-cuda12.2-cudnn8.9-ubuntu20.04-manylinux2014-python$PY_VERSION"
1817
export TF_CUDA_VERSION="12.2"
19-
export TF_CUDNN_VERSION="8.9"
2018
elif [[ "$TF_VERSION" =~ ^2\.(14)\.[0-9]+$ ]] ; then
2119
export BUILD_IMAGE="tfra/nosla-cuda11.8-cudnn8.7-ubuntu20.04-manylinux2014-python$PY_VERSION"
2220
export TF_CUDA_VERSION="11.8"
23-
export TF_CUDNN_VERSION="8.7"
2421
elif [[ "$TF_VERSION" =~ ^2\.(12|13)\.[0-9]+$ ]] ; then
2522
export BUILD_IMAGE="tfra/nosla-cuda11.8-cudnn8.6-ubuntu20.04-manylinux2014-python$PY_VERSION"
2623
export TF_CUDA_VERSION="11.8"
27-
export TF_CUDNN_VERSION="8.6"
2824
elif [[ "$TF_VERSION" =~ ^2\.([6-9]|10|11)\.[0-9]+$ ]] ; then
2925
export BUILD_IMAGE="tfra/nosla-cuda11.2-cudnn8-ubuntu20.04-manylinux2014-python$PY_VERSION"
3026
export TF_CUDA_VERSION="11.2"
31-
export TF_CUDNN_VERSION="8.1"
3227
elif [ $TF_VERSION == "2.4.1" ] ; then
3328
export BUILD_IMAGE='tfra/nosla-cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython'
3429
export TF_CUDA_VERSION="11.0"
35-
export TF_CUDNN_VERSION="8.0"
3630
elif [ $TF_VERSION == "1.15.2" ] ; then
3731
export BUILD_IMAGE='tfra/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010-multipython'
3832
export TF_CUDA_VERSION="10.0"
39-
export TF_CUDNN_VERSION="7.6"
4033
else
4134
echo "TF_VERSION is invalid: $TF_VERSION!"
4235
exit 1
4336
fi
4437

4538
echo "BUILD_IMAGE is $BUILD_IMAGE"
4639
echo "TF_CUDA_VERSION is $TF_CUDA_VERSION"
47-
echo "TF_CUDNN_VERSION is $TF_CUDNN_VERSION"
4840

4941
if [ -z $HOROVOD_VERSION ] ; then
5042
export HOROVOD_VERSION='0.28.1'
@@ -66,7 +58,6 @@ DOCKER_BUILDKIT=1 docker build --no-cache \
6658
--build-arg TF_NAME \
6759
--build-arg TF_NEED_CUDA \
6860
--build-arg TF_CUDA_VERSION \
69-
--build-arg TF_CUDNN_VERSION \
7061
--build-arg HOROVOD_VERSION \
7162
--build-arg BUILD_IMAGE \
7263
--build-arg NIGHTLY_FLAG \

0 commit comments

Comments
 (0)