added pytorch
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
maltegrosse 2023-05-09 17:03:32 +09:00
parent 91608947b4
commit f5fec02bec
1 changed files with 1 additions and 0 deletions

View File

@ -107,6 +107,7 @@ ENV NUM_GPUS=1
#==2.11.0=cuda* #==2.11.0=cuda*
#RUN mamba install -y -c conda-forge tensorflow-gpu #RUN mamba install -y -c conda-forge tensorflow-gpu
RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge
RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
WORKDIR /tmp WORKDIR /tmp
RUN mamba install --quiet --yes \ RUN mamba install --quiet --yes \
'notebook' \ 'notebook' \