abhisheksan commited on
Commit
4fbbf74
·
1 Parent(s): 0cec923

Refactor Dockerfile to create NUMBA_CACHE_DIR in /tmp with 777 permissions

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. app/main.py +2 -2
Dockerfile CHANGED
@@ -10,6 +10,8 @@ RUN apt-get update && apt-get install -y \
10
  ffmpeg
11
  COPY ./requirements.txt /code/requirements.txt
12
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 
13
 
14
  COPY . /code/
15
 
 
10
  ffmpeg
11
  COPY ./requirements.txt /code/requirements.txt
12
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
13
+ RUN mkdir -m 777 /tmp/NUMBA_CACHE_DIR
14
+ ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
15
 
16
  COPY . /code/
17
 
app/main.py CHANGED
@@ -9,8 +9,8 @@ import logging
9
  import os
10
  import tempfile
11
  os.environ['NUMBA_DISABLE_JIT'] = '1'
12
- os.environ['NUMBA_DISABLE_FUNCTION_CACHING'] = '1'
13
- os.environ['NUMBA_CACHE_DIR'] = "../temp"
14
  import numba
15
 
16
  numba.config.DISABLE_JIT = True
 
9
  import os
10
  import tempfile
11
  os.environ['NUMBA_DISABLE_JIT'] = '1'
12
+ # os.environ['NUMBA_DISABLE_FUNCTION_CACHING'] = '1'
13
+ # os.environ['NUMBA_CACHE_DIR'] = "../temp"
14
  import numba
15
 
16
  numba.config.DISABLE_JIT = True