Update TensorFlow version used in tests (NVIDIA#4739)

- 2.10->2.11.1 - 2.11->2.12 - nvidia-tensorflow=1.15.5+nv23.02 - horovod 0.26->0.27 Signed-off-by: Janusz Lisiecki <[email protected]>
huxuan · Mar 29, 2023 · a00fbe5 · a00fbe5
1 parent fc0467a
commit a00fbe5
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/docker/Dockerfile.customopbuilder.clean b/docker/Dockerfile.customopbuilder.clean
@@ -34,11 +34,25 @@ RUN rm -rf /etc/apt/sources.list.d/jonathonf-* && \
         cd Python-3.7.2 && \
         ./configure --enable-optimizations && \
         make -j8 build_all && \
-        make -j8 altinstall; \
+        make -j8 altinstall && \
+        cd .. && \
+        rm -rf Python-3.7.2; \
+    elif [ "${PYVER}" = "3.8" ]; then \
+        apt-get install -y --no-install-recommends libffi-dev && \
+        wget https://www.python.org/ftp/python/3.8.16/Python-3.8.16.tar.xz && \
+        tar -xf Python-3.8.16.tar.xz && \
+        cd Python-3.8.16 && \
+        ./configure --enable-optimizations --prefix=/usr && \
+        make -j8 build_all && \
+        make -j8 altinstall && \
+        cd .. && \
+        rm -rf Python-3.8.16; \
     else \
         if [ $(apt-cache search python$PYVER | wc -l) -eq 0 ]; then \
             apt-get install software-properties-common -y --no-install-recommends && \
+            mv /usr/bin/python3 /usr/bin/python3_bak && ln -s /usr/bin/python3.5 /usr/bin/python3 && \
             add-apt-repository ppa:deadsnakes/ppa -y && \
+            mv /usr/bin/python3_bak /usr/bin/python3 && \
             apt-get update; \
         fi && \
         curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash && \

diff --git a/qa/TL1_tensorflow-dali_test/test.sh b/qa/TL1_tensorflow-dali_test/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash -e
-pip_packages='horovod==0.26.1'
+pip_packages='horovod==0.27'
 target_dir=./docs/examples/use_cases/tensorflow/resnet-n
 
 do_once() {

diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -475,11 +475,11 @@ def get_pyvers_name(self, url, cuda_version):
                             "mxnet-cu{cuda_v}"),
                 CudaPackage("tensorflow-gpu",
                             {"110": [
-                                PckgVer("2.10.1", python_min_ver="3.7", alias="tensorflow",
+                                PckgVer("2.11.1", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24"]),
-                                PckgVer("2.11.0", python_min_ver="3.7", alias="tensorflow",
+                                PckgVer("2.12.0", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24"]),
-                                PckgVer("1.15.5+nv22.11", python_min_ver="3.8",
+                                PckgVer("1.15.5+nv23.02", python_min_ver="3.8",
                                         python_max_ver="3.8", alias="nvidia-tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24"])]}),
                 CudaPackageExtraIndex("torch",