Fix OMP_NUM_THREADS environment variable issue
Browse files- Dockerfile +4 -0
Dockerfile
CHANGED
|
@@ -2,6 +2,10 @@ FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
# Install system dependencies
|
| 6 |
RUN apt-get update && apt-get install -y \
|
| 7 |
build-essential \
|
|
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
| 5 |
+
# Fix OMP environment variable issue
|
| 6 |
+
ENV OMP_NUM_THREADS=4
|
| 7 |
+
ENV MKL_NUM_THREADS=4
|
| 8 |
+
|
| 9 |
# Install system dependencies
|
| 10 |
RUN apt-get update && apt-get install -y \
|
| 11 |
build-essential \
|