Merge pull request #765 from linshokaku/ci-ubuntu-20.04

Upgrade of ubuntu version performed by CI

Author: linshokaku

.flexci/linux/build_and_push.sh

@@ -36,7 +36,7 @@ case "${TARGET}" in
     torch110 )
         # PyTorch 1.10 + Python 3.8
         docker_build_and_push \
-            --build-arg base_image="nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04" \
+            --build-arg base_image="nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04" \
             --build-arg python_version="3.8.15" \
             --build-arg pip_install_torch_args="torch==1.10.* torchvision==0.11.* -f https://download.pytorch.org/whl/cu113/torch_stable.html" \
             --build-arg pip_install_dep_args="cupy-cuda11x pytorch-ignite onnx ${TEST_PIP_PACKAGES}"
@@ -45,25 +45,25 @@ case "${TARGET}" in
     torch111 )
         # PyTorch 1.11 + Python 3.9
         docker_build_and_push \
-            --build-arg base_image="nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04" \
+            --build-arg base_image="nvidia/cuda:11.5.2-cudnn8-devel-ubuntu20.04" \
             --build-arg python_version="3.9.7" \
-            --build-arg pip_install_torch_args="torch==1.11.* torchvision==0.12.* -f https://download.pytorch.org/whl/cu113/torch_stable.html" \
+            --build-arg pip_install_torch_args="torch==1.11.* torchvision==0.12.* -f https://download.pytorch.org/whl/cu115/torch_stable.html" \
             --build-arg pip_install_dep_args="cupy-cuda11x pytorch-ignite onnx ${TEST_PIP_PACKAGES}"
         ;;
 
     torch112 )
         # PyTorch 1.12 + Python 3.10
         docker_build_and_push \
-            --build-arg base_image="nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04" \
+            --build-arg base_image="nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04" \
             --build-arg python_version="3.10.5" \
-            --build-arg pip_install_torch_args="torch==1.12.* torchvision==0.13.* -f https://download.pytorch.org/whl/cu113/torch_stable.html" \
+            --build-arg pip_install_torch_args="torch==1.12.* torchvision==0.13.* -f https://download.pytorch.org/whl/cu117/torch_stable.html" \
             --build-arg pip_install_dep_args="cupy-cuda11x pytorch-ignite onnx ${TEST_PIP_PACKAGES}"
         ;;
 
     torch113 )
         # PyTorch 1.13 + Python 3.10
         docker_build_and_push \
-            --build-arg base_image="nvidia/cuda:11.7.1-cudnn8-devel-ubuntu18.04" \
+            --build-arg base_image="nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04" \
             --build-arg python_version="3.10.5" \
             --build-arg pip_install_torch_args="torch==1.13.* torchvision==0.14.* -f https://download.pytorch.org/whl/cu117/torch_stable.html" \
             --build-arg pip_install_dep_args="cupy-cuda11x pytorch-ignite onnx ${TEST_PIP_PACKAGES}"
@@ -72,7 +72,7 @@ case "${TARGET}" in
     torch200 )
         # PyTorch 2.0 + Python 3.10
         docker_build_and_push \
-            --build-arg base_image="nvidia/cuda:11.7.1-cudnn8-devel-ubuntu18.04" \
+            --build-arg base_image="nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04" \
             --build-arg python_version="3.10.5" \
             --build-arg pip_install_torch_args="torch==2.0.* torchvision==0.15.* -f https://download.pytorch.org/whl/cu117/torch_stable.html" \
             --build-arg pip_install_dep_args="cupy-cuda11x pytorch-ignite onnx ${TEST_PIP_PACKAGES}"