mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2024-11-21 01:16:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

add gpu docker

Browse Source
This commit is contained in:
pengluan 2022-04-26 13:19:09 +08:00
parent a2aef71c1c
commit af1b39460d
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
images/ubuntu-gpu/build.sh
View File

@ -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