diff --git a/Dockerfile.moderngpu b/Dockerfile.moderngpu new file mode 100644 index 00000000..b505e28b --- /dev/null +++ b/Dockerfile.moderngpu @@ -0,0 +1,31 @@ + +FROM nvidia/cuda:11.8.0-base-ubuntu20.04 + +ENV PYTHON_VERSION=3.8 + +RUN export DEBIAN_FRONTEND=noninteractive \ + && apt-get -qq update \ + && apt-get -qq install --no-install-recommends \ + libsndfile1-dev \ + git \ + python${PYTHON_VERSION} \ + python${PYTHON_VERSION}-venv \ + python3-pip \ + && rm -rf /var/lib/apt/lists/* + +RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \ + ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \ + ln -s -f /usr/bin/pip3 /usr/bin/pip + +RUN pip install --upgrade pip + +# 2. Copy files +COPY . /src + +RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 + +WORKDIR /src +# 3. Install dependencies +RUN pip install -r requirements-docker.txt + +RUN python3 setup.py install