We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 71a1d50 commit 6c0345eCopy full SHA for 6c0345e
docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0
@@ -86,7 +86,7 @@ RUN git clone --recursive --branch v${PYTORCH_VERSION} https://github.com/pytorc
86
export USE_CUDA=1 USE_CUDNN=1 USE_MKLDNN=0 USE_DISTRIBUTED=0 BUILD_TEST=0 && \
87
export TORCH_CUDA_ARCH_LIST="8.7" CUDA_HOME=/usr/local/cuda && \
88
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} && \
+ export MAX_JOBS=${MAX_JOBS} PYTORCH_BUILD_VERSION=${PYTORCH_VERSION} PYTORCH_BUILD_NUMBER=1 && \
90
python3 setup.py bdist_wheel && \
91
python3 -m pip install dist/torch-*.whl
92
0 commit comments