nroggendorff commited on
Commit
e6ad1f5
·
verified ·
1 Parent(s): c1b1497

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -21,6 +21,10 @@ ENV PYTHONUNBUFFERED=1
21
  ENV TQDM_POSITION=-1
22
  ENV TQDM_MININTERVAL=1
23
 
 
 
 
 
24
  RUN mkdir /.cache && chmod 777 /.cache && chmod 777 /app
25
 
26
  CMD ["python3", "train.py"]
 
21
  ENV TQDM_POSITION=-1
22
  ENV TQDM_MININTERVAL=1
23
 
24
+ ENV OPENBLAS_NUM_THREADS=1
25
+ ENV MKL_NUM_THREADS=1
26
+ ENV OMP_NUM_THREADS=1
27
+
28
  RUN mkdir /.cache && chmod 777 /.cache && chmod 777 /app
29
 
30
  CMD ["python3", "train.py"]