Update Dockerfile
Browse files- Dockerfile +8 -1
Dockerfile
CHANGED
@@ -22,10 +22,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
22 |
&& rm -rf /var/lib/apt/lists/*
|
23 |
|
24 |
# Create a non-root user
|
|
|
25 |
RUN useradd -m -s /bin/bash appuser && \
|
26 |
mkdir -p /home/appuser/.cache/torch && \
|
27 |
mkdir -p /home/appuser/.triton && \
|
28 |
-
chown -R appuser:appuser /home/appuser
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
# Set cache directories environment variables
|
31 |
ENV TORCH_HOME=/home/appuser/.cache/torch \
|
|
|
22 |
&& rm -rf /var/lib/apt/lists/*
|
23 |
|
24 |
# Create a non-root user
|
25 |
+
# In your Dockerfile:
|
26 |
RUN useradd -m -s /bin/bash appuser && \
|
27 |
mkdir -p /home/appuser/.cache/torch && \
|
28 |
mkdir -p /home/appuser/.triton && \
|
29 |
+
- chown -R appuser:appuser /home/appuser
|
30 |
+
+ chown -R appuser:appuser /home/appuser && \
|
31 |
+
+ chmod 755 /home/appuser/.triton
|
32 |
+
|
33 |
+
# Add these ENV variables
|
34 |
+
ENV TRITON_CACHE_DIR=/home/appuser/.triton \
|
35 |
+
TORCH_INDUCTOR_CACHE_DIR=/home/appuser/.triton
|
36 |
|
37 |
# Set cache directories environment variables
|
38 |
ENV TORCH_HOME=/home/appuser/.cache/torch \
|