Trying different image

prokube-ai · Nov 1, 2024 · d4b96ec · d4b96ec
1 parent fb1c41c
commit d4b96ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/llama2-chat-predictor/Dockerfile b/images/llama2-chat-predictor/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:24.10-py3
+FROM pytorch/pytorch:latest
 RUN apt-get update && apt-get install -y libglib2.0-0
 RUN mkdir /app
 WORKDIR /app