forked from Alec-Wright/Automated-GuitarAmpModelling
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Dockerfile_pytorch
31 lines (23 loc) · 1.21 KB
/
Dockerfile_pytorch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# Copyright (c) 2023 Aida DSP (aidadsp.cc)
#
# Source code courtesy of Massimo Pennazio <maxipenna@libero.it>
#
# Example usage
# build: docker build -f Dockerfile_pytorch --build-arg host_uid=1000 --build-arg host_gid=1000 . -t pytorch
# run: docker run --gpus all -v $PWD:/workdir:rw -w /workdir -it -p '127.0.0.1:8888:8888' pytorch:latest
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
LABEL maintainer "Massimo Pennazio <maxipenna@libero.it>"
WORKDIR /workdir
RUN pip3 --disable-pip-version-check --no-cache-dir install scipy==1.10.1 tensorboard==2.11.0 tensorflow==2.11.0 jupyterlab==3.6.1 librosa==0.10.0 auraloss==0.4.0
RUN pip3 --disable-pip-version-check --no-cache-dir install noisereduce==2.0.1
RUN pip3 --disable-pip-version-check --no-cache-dir install torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
ENV USER_NAME aidadsp
# Create a user with same uid and gid of the current host's user
ARG host_uid=1000
ARG host_gid=1000
RUN groupadd -g $host_gid $USER_NAME && \
useradd -g $host_gid -m -s /bin/bash -u $host_uid $USER_NAME
# From this line below all commands run as user
USER $USER_NAME
ENTRYPOINT ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser"]
EXPOSE 8888