tezuesh commited on
Commit
1de92f4
·
verified ·
1 Parent(s): 43ce8b7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 \