[#36] Set OMP_NUM_THREADS=1 to turn off the openblas threading

More info: ultralytics/yolov5#2863
xserrat · Nov 19, 2023 · f87c2d3 · f87c2d3
1 parent cefe80d
commit f87c2d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -3,6 +3,7 @@ FROM nvidia/cuda:11.8.0-base-ubuntu22.04
 
 USER root
 ENV TORCH_HOME=/data/models
+ENV OMP_NUM_THREADS=1
 
 # Install required tools
 # Notes: