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

This commit is contained in:
maltegrosse 2023-02-01 10:18:25 +01:00
parent 50c5f3807d
commit d51e6fa5b1
1 changed files with 3 additions and 2 deletions

View File

@ -100,4 +100,5 @@ RUN set -x && \
#RUN CONDA_CUDA_OVERRIDE="11.2" mamba install --yes -c conda-forge tensorflow==2.10.0=cuda112py310* #RUN CONDA_CUDA_OVERRIDE="11.2" mamba install --yes -c conda-forge tensorflow==2.10.0=cuda112py310*
ENV CONDA_CUDA_OVERRIDE="11.2" ENV CONDA_CUDA_OVERRIDE="11.2"
RUN mamba install -y -c conda-forge tensorflow-gpu RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
#RUN mamba install -y -c conda-forge tensorflow-gpu