Martí Umbert
commited on
Commit
·
a372da6
1
Parent(s):
d4b369d
Dockerfile: numba_cache_dir
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -52,10 +52,12 @@ COPY . .
|
|
| 52 |
|
| 53 |
ENV MPLCONFIGDIR=/var/cache/matplotlib
|
| 54 |
ENV TRANSFORMERS_CACHE=/var/cache/huggingface/hub
|
|
|
|
| 55 |
RUN mkdir -p /var/cache/matplotlib && chmod -R 777 /var/cache/matplotlib && \
|
| 56 |
mkdir -p /var/cache/huggingface/hub && chmod -R 777 /var/cache/huggingface/ && chmod -R 777 /var/cache/huggingface/hub && \
|
| 57 |
mkdir /.config && chmod -R 777 /.config && \
|
| 58 |
-
mkdir /nltk_data && chmod -R 777 /nltk_data
|
|
|
|
| 59 |
|
| 60 |
RUN python -c 'import torch; from dcase24t6.nn.hub import baseline_pipeline; model = baseline_pipeline()'
|
| 61 |
|
|
|
|
| 52 |
|
| 53 |
ENV MPLCONFIGDIR=/var/cache/matplotlib
|
| 54 |
ENV TRANSFORMERS_CACHE=/var/cache/huggingface/hub
|
| 55 |
+
ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
|
| 56 |
RUN mkdir -p /var/cache/matplotlib && chmod -R 777 /var/cache/matplotlib && \
|
| 57 |
mkdir -p /var/cache/huggingface/hub && chmod -R 777 /var/cache/huggingface/ && chmod -R 777 /var/cache/huggingface/hub && \
|
| 58 |
mkdir /.config && chmod -R 777 /.config && \
|
| 59 |
+
mkdir /nltk_data && chmod -R 777 /nltk_data && \
|
| 60 |
+
mkdir -p /tmp/NUMBA_CACHE_DIR && chmod -R 777 /tmp/NUMBA_CACHE_DIR
|
| 61 |
|
| 62 |
RUN python -c 'import torch; from dcase24t6.nn.hub import baseline_pipeline; model = baseline_pipeline()'
|
| 63 |
|