Less?

Author: jrbourbeau

Dockerfile

@@ -12,8 +12,6 @@ RUN mamba env update -n base --file pytorch.yml \
     && conda clean -afy \
     && mamba uninstall -y pytorch torchvision \
     && mamba install -y -n base -c conda-forge \
-        "cudatoolkit=${CUDA_VER%.*}.*" \
-        "cuda-version=${CUDA_VER%.*}.*" \
         pytorch-gpu \
         torchvision \
     && conda clean -afy \