cuda 12.1
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2024-03-18 14:20:38 +09:00
parent 82acfa910f
commit fecc1cbd88
1 changed files with 2 additions and 2 deletions

View File

@ -3,6 +3,6 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d
ARG NB_USER="jovyan"
ARG NB_UID="1000"
ARG NB_GID="100"
ENV CONDA_CUDA_OVERRIDE="11.2"
ENV CONDA_CUDA_OVERRIDE="12.1"
ENV NUM_GPUS=1
RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia