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

@ -99,5 +99,6 @@ RUN set -x && \
#RUN conda install -c conda-forge tensorflow-gpu=2.11 -y #RUN conda install -c conda-forge tensorflow-gpu=2.11 -y
#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