Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 12 additions & 6 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -962,7 +962,6 @@ def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap):
libcurl-devel \\
libb64-devel \\
gperftools-devel \\
patchelf \\
python3-pip \\
python3-setuptools \\
rapidjson-devel \\
Expand Down Expand Up @@ -990,7 +989,8 @@ def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap):
wheel \\
setuptools \\
docker \\
virtualenv
virtualenv \\
patchelf==0.17.2

# Install boost version >= 1.78 for boost::span
# Current libboost-dev apt packages are < 1.78, so install from tar.gz
Expand Down Expand Up @@ -1089,7 +1089,6 @@ def create_dockerfile_buildbase(ddir, dockerfile_name, argmap):
libcurl4-openssl-dev \\
libb64-dev \\
libgoogle-perftools-dev \\
patchelf \\
python3-dev \\
python3-pip \\
python3-wheel \\
Expand All @@ -1110,7 +1109,8 @@ def create_dockerfile_buildbase(ddir, dockerfile_name, argmap):
RUN pip3 install --upgrade \\
build \\
docker \\
virtualenv
virtualenv \\
patchelf==0.17.2

# Install boost version >= 1.78 for boost::span
# Current libboost-dev apt packages are < 1.78, so install from tar.gz
Expand Down Expand Up @@ -1354,11 +1354,14 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
libcurl-devel \\
libb64-devel \\
gperftools-devel \\
patchelf \\
wget \\
python3-pip \\
numactl-devel

RUN pip3 install --upgrade pip \\
&& pip3 install --upgrade \\
patchelf==0.17.2

"""
else:
df += """
Expand Down Expand Up @@ -1562,7 +1565,10 @@ def add_cpu_libs_to_linux_dockerfile(backends, target_machine):

# patchelf is needed to add deps of libcublasLt.so.12 to libtorch_cuda.so
RUN apt-get update \\
&& apt-get install -y --no-install-recommends openmpi-bin patchelf
&& apt-get install -y --no-install-recommends openmpi-bin
RUN pip3 install --upgrade pip \\
&& pip3 install --upgrade \\
patchelf==0.17.2

ENV LD_LIBRARY_PATH /usr/local/cuda/targets/{cuda_arch}-linux/lib:/usr/local/cuda/lib64/stubs:${{LD_LIBRARY_PATH}}
""".format(
Expand Down
3 changes: 1 addition & 2 deletions docs/user_guide/jetson.md
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,6 @@ apt-get update && \
libtool \
libboost-dev \
rapidjson-dev \
patchelf \
pkg-config \
libopenblas-dev \
libarchive-dev \
Expand All @@ -95,7 +94,7 @@ apt-get update && \
Additional Onnx Runtime dependencies must be installed to build the Onnx Runtime backend:

```
pip3 install --upgrade flake8 flatbuffers
pip3 install --upgrade flake8 flatbuffers patchelf==0.17.2
```

Additional PyTorch dependencies must be installed to build (and run) the PyTorch backend:
Expand Down
Loading