Update Dockerfile
Browse files- 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 \
|