diff --git a/images/ubuntu-gpu/build.sh b/images/ubuntu-gpu/build.sh index 24abe6c3..10853dfd 100644 --- a/images/ubuntu-gpu/build.sh +++ b/images/ubuntu-gpu/build.sh @@ -3,11 +3,9 @@ hubhost=ai.tencentmusic.com/tme-public base_image=nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 docker build -t $hubhost/ubuntu-gpu:cuda10.2-cudnn7 --build-arg FROM_IMAGES=$base_image -f cuda/Dockerfile . -docker build -t $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.6 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda10.2-cudnn7 --build-arg PYTHON_VERSION=python3.6 -f cuda/python/Dockerfile . docker build -t $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.7 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda10.2-cudnn7 --build-arg PYTHON_VERSION=python3.7 -f cuda/python/Dockerfile . docker build -t $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.8 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda10.2-cudnn7 --build-arg PYTHON_VERSION=python3.8 -f cuda/python/Dockerfile . docker push $hubhost/ubuntu-gpu:cuda10.2-cudnn7 -docker push $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.6 docker push $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.7 docker push $hubhost/ubuntu-gpu:cuda10.2-cudnn7-python3.8