MogensR commited on
Commit
7c1ea91
·
verified ·
1 Parent(s): 4f169e6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -13,7 +13,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
13
  TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" \
14
  CUDA_VISIBLE_DEVICES="0" \
15
  OMP_NUM_THREADS=2 \
16
- RUN export OMP_NUM_THREADS=2
17
  OPENBLAS_NUM_THREADS=1 \
18
  MKL_NUM_THREADS=1 \
19
  NUMEXPR_NUM_THREADS=1 \
 
13
  TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" \
14
  CUDA_VISIBLE_DEVICES="0" \
15
  OMP_NUM_THREADS=2 \
16
+ RUN export OMP_NUM_THREADS=2
17
  OPENBLAS_NUM_THREADS=1 \
18
  MKL_NUM_THREADS=1 \
19
  NUMEXPR_NUM_THREADS=1 \