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

fix
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2024-03-18 15:23:07 +09:00
parent 707b6ec6af
commit dc259d2af9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -9,7 +9,7 @@ USER root
RUN apt-get update --yes && \ RUN apt-get update --yes && \
apt-get install --yes --no-install-recommends \ apt-get install --yes --no-install-recommends \
ffmpeg \ ffmpeg \
libportaudio2 \ libportaudio2 && \
apt-get clean && rm -rf /var/lib/apt/lists/* apt-get clean && rm -rf /var/lib/apt/lists/*
USER $NB_UID USER $NB_UID
RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia