Skip to content

Commit 6c0345e

Browse files
committed
Fix PyTorch build: add PYTORCH_BUILD_NUMBER env var
1 parent 71a1d50 commit 6c0345e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ RUN git clone --recursive --branch v${PYTORCH_VERSION} https://github.com/pytorc
8686
export USE_CUDA=1 USE_CUDNN=1 USE_MKLDNN=0 USE_DISTRIBUTED=0 BUILD_TEST=0 && \
8787
export TORCH_CUDA_ARCH_LIST="8.7" CUDA_HOME=/usr/local/cuda && \
8888
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu CUDNN_INCLUDE_DIR=/usr/include && \
89-
export MAX_JOBS=${MAX_JOBS} PYTORCH_BUILD_VERSION=${PYTORCH_VERSION} && \
89+
export MAX_JOBS=${MAX_JOBS} PYTORCH_BUILD_VERSION=${PYTORCH_VERSION} PYTORCH_BUILD_NUMBER=1 && \
9090
python3 setup.py bdist_wheel && \
9191
python3 -m pip install dist/torch-*.whl
9292

0 commit comments

Comments
 (0)