[push-ci-image]

huggingface · Apr 29, 2024 · cf0a318 · cf0a318
1 parent 3d38101
commit cf0a318
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 5 deletions.
diff --git a/docker/examples-torch.dockerfile b/docker/examples-torch.dockerfile
@@ -16,8 +16,11 @@ RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 RUN pip cache remove "nvidia-*"
 RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true
+RUN pip uninstall -y `pip freeze | grep "triton-*"` || true
+
+
+
 RUN pip cache remove triton
 RUN apt-get --purge remove "*nvidia*" || true
-RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*"  || true
 RUN apt-get autoremove
 RUN apt-get autoclean
diff --git a/docker/pipeline-torch.dockerfile b/docker/pipeline-torch.dockerfile
@@ -16,8 +16,11 @@ RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 RUN pip cache remove "nvidia-*"
 RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true
+RUN pip uninstall -y `pip freeze | grep "triton-*"` || true
+
+
+
 RUN pip cache remove triton
 RUN apt-get --purge remove "*nvidia*" || true
-RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*"  || true
 RUN apt-get autoremove
 RUN apt-get autoclean
diff --git a/docker/torch-jax-light.dockerfile b/docker/torch-jax-light.dockerfile
@@ -17,8 +17,11 @@ RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 RUN pip cache remove "nvidia-*"
 RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true
+RUN pip uninstall -y `pip freeze | grep "triton-*"` || true
+
+
+
 RUN pip cache remove triton
 RUN apt-get --purge remove "*nvidia*" || true
-RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*"  || true
 RUN apt-get autoremove
 RUN apt-get autoclean
diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile
@@ -16,8 +16,11 @@ RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 RUN pip cache remove "nvidia-*"
 RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true
+RUN pip uninstall -y `pip freeze | grep "triton-*"` || true
+
+
+
 RUN pip cache remove triton
 RUN apt-get --purge remove "*nvidia*" || true
-RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*"  || true
 RUN apt-get autoremove
 RUN apt-get autoclean
diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile
@@ -18,8 +18,11 @@ RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 RUN pip cache remove "nvidia-*"
 RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true
+RUN pip uninstall -y `pip freeze | grep "triton-*"` || true
+
+
+
 RUN pip cache remove triton
 RUN apt-get --purge remove "*nvidia*" || true
-RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*"  || true
 RUN apt-get autoremove
 RUN apt-get autoclean