Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -35,7 +35,7 @@ RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu12
|
|
| 35 |
SHELL ["/bin/bash","-lc"]
|
| 36 |
ARG SM="8.9"
|
| 37 |
ENV TORCH_CUDA_ARCH_LIST=${SM}
|
| 38 |
-
ENV MAX_JOBS=
|
| 39 |
|
| 40 |
RUN ABI=$(python -c "import torch; print(int(torch._C._GLIBCXX_USE_CXX11_ABI))") \
|
| 41 |
&& export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=${ABI}" \
|
|
@@ -51,7 +51,7 @@ PY
|
|
| 51 |
# Apex (build do fonte com extensões C++/CUDA, conforme manual)
|
| 52 |
# ------------------------------------------------
|
| 53 |
ENV CUDA_HOME=/usr/local/cuda
|
| 54 |
-
ENV NVCC_APPEND_FLAGS="--threads
|
| 55 |
ENV APEX_PARALLEL_BUILD=2
|
| 56 |
ENV APEX_CPP_EXT=1
|
| 57 |
ENV APEX_CUDA_EXT=1
|
|
|
|
| 35 |
SHELL ["/bin/bash","-lc"]
|
| 36 |
ARG SM="8.9"
|
| 37 |
ENV TORCH_CUDA_ARCH_LIST=${SM}
|
| 38 |
+
ENV MAX_JOBS=8
|
| 39 |
|
| 40 |
RUN ABI=$(python -c "import torch; print(int(torch._C._GLIBCXX_USE_CXX11_ABI))") \
|
| 41 |
&& export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=${ABI}" \
|
|
|
|
| 51 |
# Apex (build do fonte com extensões C++/CUDA, conforme manual)
|
| 52 |
# ------------------------------------------------
|
| 53 |
ENV CUDA_HOME=/usr/local/cuda
|
| 54 |
+
ENV NVCC_APPEND_FLAGS="--threads 2"
|
| 55 |
ENV APEX_PARALLEL_BUILD=2
|
| 56 |
ENV APEX_CPP_EXT=1
|
| 57 |
ENV APEX_CUDA_EXT=1
|