jianbing/cuda repository overview

# sudo docker build -t jianbing/cuda:12.2.0-devel-ubuntu20.04-python3-apt -f Dockerfile .

FROM nvidia/cuda:12.2.0-devel-ubuntu20.04

RUN sed -i "s/archive.ubuntu.com/192.168.5.142:8081\/repository/g" /etc/apt/sources.list && sed -i "s/security.ubuntu.com/192.168.5.142:8081\/repository/g" /etc/apt/sources.list
RUN apt update -y && apt install python3 -y && apt install python3-pip -y

RUN pip install -i http://192.168.5.142:8081/repository/pypi/simple --trusted-host 192.168.5.142 --timeout 3000 transformers==4.30.2 accelerate sentencepiece astunparse deepspeed
RUN pip install -i http://192.168.5.142:8081/repository/pypi/simple --trusted-host 192.168.5.142 --timeout 3000 protobuf transformers==4.30.2 cpm_kernels torch>=2.0 gradio==3.39 mdtex2html sentencepiece accelerate sse-starlette streamlit>=1.24.0 fastapi==0.95.1 typing_extensions==4.4.0 uvicorn sse_starlette loguru

# 设置pyton3 python
RUN ln -s /usr/bin/python3.8 /usr/bin/python

# sudo docker build -t jianbing/cuda:12.2.0-devel-ubuntu20.04-python-3.10.12 .

FROM nvidia/cuda:12.2.0-devel-ubuntu20.04

RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
  && echo "Asia/Shanghai" > /etc/timezone

RUN sed -i "s/archive.ubuntu.com/192.168.5.142:8081\/repository/g" /etc/apt/sources.list \
  && sed -i "s/security.ubuntu.com/192.168.5.142:8081\/repository/g" /etc/apt/sources.list

RUN apt update -y && apt upgrade -y

ADD Python-3.10.12.tgz /usr/src/
RUN apt update -y && apt upgrade -y && \
    apt-get install -y wget build-essential checkinstall  libreadline-gplv2-dev  libncursesw5-dev  libssl-dev  libsqlite3-dev tk-dev libgdbm-dev libc6-dev libbz2-dev libffi-dev zlib1g-dev && \
    cd /usr/src/Python-3.10.12 && \
    ./configure --enable-optimizations && \
    make altinstall

RUN ln -s /usr/local/bin/python3.10 /usr/local/bin/python \
    && ln -s /usr/local/bin/python3.10 /usr/local/bin/python3 \
    && ln -s /usr/local/bin/pip3.10 /usr/local/bin/pip \
    && ln -s /usr/local/bin/pip3.10 /usr/local/bin/pip3

# 设置pip私服
RUN pip config set global.index-url http://192.168.5.142:8081/repository/pypi/simple \
    && pip config set global.trusted-host 192.168.5.142

RUN python -m pip install --upgrade pip
# RUN pip3 install torch torchvision torchaudio
RUN pip install torch torchvision torchaudio