Fix attempt 6

pharmbio · Sep 17, 2024 · 54f6dd6 · 54f6dd6
1 parent 0971997
commit 54f6dd6
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/docker/env.Dockerfile b/docker/env.Dockerfile
@@ -69,13 +69,13 @@ RUN <<EOF
                         torchaudio;
                 python3 -m pip install --no-cache-dir \ 
                         pytorch_toolbelt;
-                python3 -m pip install --no-cache-dir -f https://data.pyg.org/whl/torch-2.2.0+cu121.html \
-                        pyg-lib \
-                        torch-scatter \
-                        torch-sparse \
-                        torch-cluster \
-                        torch-spline-conv \
-                        torch-geometric;
+                # python3 -m pip install --no-cache-dir -f https://data.pyg.org/whl/torch-2.2.0+cu121.html \
+                #         pyg-lib \
+                #         torch-scatter \
+                #         torch-sparse \
+                #         torch-cluster \
+                #         torch-spline-conv \
+                #         torch-geometric;
                 python3 -m pip install \
                         --extra-index-url=https://pypi.nvidia.com \
                         cudf-cu12==24.6.* dask-cudf-cu12==24.6.* cuml-cu12==24.6.* \
@@ -90,13 +90,13 @@ RUN <<EOF
                         torchaudio;
                 python3 -m pip install --no-cache-dir 
                         pytorch_toolbelt ;
-                python3 -m pip install --no-cache-dir -f https://data.pyg.org/whl/torch-2.2.0+cpu.html \
-                        pyg-lib \
-                        torch-scatter \
-                        torch-sparse \
-                        torch-cluster \
-                        torch-spline-conv \
-                        torch-geometric; 
+                # python3 -m pip install --no-cache-dir -f https://data.pyg.org/whl/torch-2.2.0+cpu.html \
+                #         pyg-lib \
+                #         torch-scatter \
+                #         torch-sparse \
+                #         torch-cluster \
+                #         torch-spline-conv \
+                #         torch-geometric; 
         else 
         echo "Error: unknown hardware" 
         exit 1