Skip to content

Commit

Permalink
Merge pull request #765 from linshokaku/ci-ubuntu-20.04
Browse files Browse the repository at this point in the history
Upgrade of ubuntu version performed by CI
  • Loading branch information
linshokaku authored Oct 19, 2023
2 parents efe0cee + c97550e commit e5ef589
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions .flexci/linux/build_and_push.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand All @@ -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}"
Expand All @@ -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}"
Expand Down

0 comments on commit e5ef589

Please sign in to comment.