mac9087 commited on
Commit
ef15b91
·
verified ·
1 Parent(s): d43322e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -5,6 +5,7 @@ WORKDIR /app
5
  ENV HF_HOME=/tmp/huggingface
6
  ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
7
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
 
8
  ENV PYTHONDONTWRITEBYTECODE=1
9
  ENV PYTHONUNBUFFERED=1
10
  ENV CUDA_VISIBLE_DEVICES=""
 
5
  ENV HF_HOME=/tmp/huggingface
6
  ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
7
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
8
+ ENV NUMBA_CACHE_DIR=/tmp/numba_cache
9
  ENV PYTHONDONTWRITEBYTECODE=1
10
  ENV PYTHONUNBUFFERED=1
11
  ENV CUDA_VISIBLE_DEVICES=""