cuda
ci/woodpecker/push/woodpecker Pipeline was successful
Details
ci/woodpecker/push/woodpecker Pipeline was successful
Details
This commit is contained in:
parent
50c5f3807d
commit
d51e6fa5b1
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue