diff --git a/Dockerfile b/Dockerfile index 2532d17..4003f79 100644 --- a/Dockerfile +++ b/Dockerfile @@ -100,7 +100,7 @@ RUN set -x && \ #RUN CONDA_CUDA_OVERRIDE="11.2" mamba install --yes -c conda-forge tensorflow==2.10.0=cuda112py310* ENV CONDA_CUDA_OVERRIDE="11.2" -RUN export CONDA_CUDA_OVERRIDE="11.2" +RUN nvidia-smi #RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge #==2.11.0=cuda* RUN mamba install -y -c conda-forge tensorflow-gpu