FROM  nvidia/cuda:xx.xx.x-yyyy-zzzzzz

WORKDIR /training

RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo "Asia/Shanghai" > /etc/timezone && \
    cp /etc/apt/sources.list /etc/apt/sources.list.bak && \
    sed -i 's#archive.ubuntu.com#mirrors.tuna.tsinghua.edu.cn#g' /etc/apt/sources.list && \
    sed -i 's#security.ubuntu.com#mirrors.tuna.tsinghua.edu.cn#g' /etc/apt/sources.list && \
    mkdir -p ~/.pip && touch ~/.pip/pip.conf && echo "[global]" > ~/.pip/pip.conf && echo "index-url = https://pypi.tuna.tsinghua.edu.cn/simple" >> ~/.pip/pip.conf && \
    apt-get update -y && apt-get install vim python3 python3-pip curl libgl1 libglib2.0-0 git tmux zsh -y  && apt-get clean && \
    curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py

EXPOSE 7860

docker run --name training -itd -p 7860:7860 -v /f/AI/TrainingData:/training/Data -v /f/AI/Models/:/training/Models --gpus all crazybun/cuda:12.1.0-runtime-ubuntu20.04