dockerfile init

This commit is contained in:
psr 2023-04-03 22:15:51 -07:00
parent 0440eac2bc
commit c6956334e3

27
Dockerfile Normal file
View File

@ -0,0 +1,27 @@
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive
ARG TZ=America/Los_Angeles
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update
RUN apt install -y curl wget git
RUN adduser --disabled-password --gecos '' --shell /bin/bash user
USER user
ENV HOME=/home/user
WORKDIR $HOME
RUN mkdir $HOME/.cache $HOME/.config && chmod -R 777 $HOME
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py39_23.1.0-1-Linux-x86_64.sh
RUN chmod +x Miniconda3-py39_23.1.0-1-Linux-x86_64.sh
RUN ./Miniconda3-py39_23.1.0-1-Linux-x86_64.sh -b -p /home/user/miniconda
ENV PATH="$HOME/miniconda/bin:$PATH"
RUN conda init
RUN conda install python=3.9.13
RUN python3 -m pip install --upgrade pip
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
# RUN git clone https://git.ecker.tech/mrq/ai-voice-cloning.git
# WORKDIR /home/user/ai-voice-cloning
# ENV VIRTUAL_ENV=/home/user/ai-voice-cloning/venv
# ENV PATH="$VIRTUAL_ENV/bin:$PATH"
# RUN ./setup-cuda.sh
CMD ["python"]