grosse
/
jupyterlab-datascience-gpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add audio libs
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Malte Grosse 2024-03-18 15:02:01 +09:00
parent fecc1cbd88
commit e54e96db1c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@ -5,4 +5,10 @@ ARG NB_UID="1000"
ARG NB_GID="100"
ENV CONDA_CUDA_OVERRIDE="12.1"
ENV NUM_GPUS=1
USER root
RUN apt-get install --yes --no-install-recommends \
ffmpeg \
libportaudio2 \
apt-get clean && rm -rf /var/lib/apt/lists/*
USER $NB_UID
RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia