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

添加支持cube11 gpu镜像

Browse Source
This commit is contained in:
pengluan 2022-06-24 11:40:48 +08:00
parent 884de4dd6c
commit a7e0c787e5
3 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,10 +1,10 @@
set -ex
hubhost=ccr.ccs.tencentyun.com/cube-studio
base_image=nvidia/cuda:11.0.3-cudnn8-devel-ubuntu20.04
base_image=nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04
docker build -t $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8 --build-arg FROM_IMAGES=$base_image -f cuda/Dockerfile .
docker build -t $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8-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:cuda11.0.3-cudnn8-python3.8 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda10.2-cudnn7 --build-arg PYTHON_VERSION=python3.8 -f cuda/python/Dockerfile .
docker build -t $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8-python3.7 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda11.0.3-cudnn8 --build-arg PYTHON_VERSION=python3.7 -f cuda/python/Dockerfile .
docker build -t $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8-python3.8 --build-arg FROM_IMAGES=$hubhost/ubuntu-gpu:cuda11.0.3-cudnn8 --build-arg PYTHON_VERSION=python3.8 -f cuda/python/Dockerfile .
docker push $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8
docker push $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8-python3.7
docker push $hubhost/ubuntu-gpu:cuda11.0.3-cudnn8-python3.8

3
images/ubuntu-gpu/cuda/Dockerfile
View File

@ -2,6 +2,9 @@ ARG FROM_IMAGES
FROM $FROM_IMAGES
RUN apt update
ENV TZ=Asia/Shanghai
ENV DEBIAN_FRONTEND=noninteractive
# 安装运维工具
RUN apt install -y --force-yes --no-install-recommends software-properties-common vim apt-transport-https gnupg2 ca-certificates-java rsync jq wget git dnsutils iputils-ping net-tools curl mysql-client locales zip unzip

2
images/ubuntu-gpu/cuda/python/Dockerfile
View File

@ -5,7 +5,7 @@ ARG PYTHON_VERSION
# 安装python
RUN add-apt-repository -y ppa:deadsnakes/ppa && apt update && apt install -y libsasl2-dev libpq-dev python3-pip python3-distutils
RUN set -x; apt install -y ${PYTHON_VERSION} && rm -rf /usr/bin/python && ln -s /usr/bin/${PYTHON_VERSION} /usr/bin/python
RUN set -x; rm -rf /usr/bin/python; apt install -y --fix-missing ${PYTHON_VERSION} && ln -s /usr/bin/${PYTHON_VERSION} /usr/bin/python
RUN bash -c "wget https://bootstrap.pypa.io/get-pip.py && python get-pip.py --ignore-installed" \
&& rm -rf /usr/bin/pip && ln -s /usr/bin/pip3 /usr/bin/pip