From 68406c9f0343b167014654c65010d46d51fb14e8 Mon Sep 17 00:00:00 2001
From: James Bourbeau <jrbourbeau@gmail.com>
Date: Mon, 12 May 2025 12:18:55 -0500
Subject: [PATCH] Fix GPU docker image

---
 Dockerfile | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 8b2fd39..4c9e949 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,14 +10,6 @@ WORKDIR /home
 COPY pytorch.yml pytorch.yml
 RUN mamba env update -n base --file pytorch.yml \
     && conda clean -afy \
-    && mamba uninstall -y pytorch torchvision \
-    && mamba install -y -n base -c pytorch -c nvidia -c conda-forge \
-        "cudatoolkit=${CUDA_VER%.*}.*" \
-        "cuda-version=${CUDA_VER%.*}.*" \
-        pytorch \
-        torchvision \
-        "pytorch-cuda=${CUDA_VER%.*}.*" \
-    && conda clean -afy \
     && rm pytorch.yml