CUDA Dockerfile working

argilla-io · May 14, 2024 · a44034c · a44034c
1 parent e5b3cd4
commit a44034c
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/CUDA.Dockerfile → docker/CUDA.Dockerfile b/CUDA.Dockerfile → docker/CUDA.Dockerfile
@@ -1,23 +1,18 @@
-# Use the specified base image
 FROM runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04 AS base
 
-# Run system updates and clean up
 RUN apt-get update && \
     apt-get upgrade -y && \
     apt-get autoremove -y && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-# Set the working directory to /
-WORKDIR /app
+WORKDIR /
 
 FROM runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04
 
-# RUN pip install "distilabel[argilla,hf-transformers,hf-inference-endpoints,llama-cpp,vllm]" openai
-
 COPY . .
 
-RUN pip install -e .
+RUN pip install -e ".[argilla,hf-transformers,hf-inference-endpoints,llama-cpp,vllm]"
 
 EXPOSE 80