Spaces:
Runtime error
Runtime error
Soumic
commited on
Commit
•
4792fc5
1
Parent(s):
76baca9
:lady_beetle: Set TRANSFORMERS_CACHE to writable directory
Browse files- Dockerfile +6 -0
Dockerfile
CHANGED
@@ -4,6 +4,12 @@ FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
|
|
4 |
# Set a working directory in the container
|
5 |
WORKDIR /workspace
|
6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
# Install Python dependencies
|
8 |
COPY requirements.txt .
|
9 |
RUN pip install --upgrade pip
|
|
|
4 |
# Set a working directory in the container
|
5 |
WORKDIR /workspace
|
6 |
|
7 |
+
# Create a writable cache directory
|
8 |
+
RUN mkdir -p /workspace/cache
|
9 |
+
|
10 |
+
# Set the environment variable for cache
|
11 |
+
ENV TRANSFORMERS_CACHE=/workspace/cache
|
12 |
+
|
13 |
# Install Python dependencies
|
14 |
COPY requirements.txt .
|
15 |
RUN pip install --upgrade pip
|