Skip to content

Commit

Permalink
PIP tune cache
Browse files Browse the repository at this point in the history
  • Loading branch information
YanWenKun committed Jun 30, 2024
1 parent 6221350 commit bdb9a45
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
4 changes: 1 addition & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -91,9 +91,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
RUN --mount=type=cache,target=/root/.cache/pip \
pip uninstall --break-system-packages --yes \
onnxruntime-gpu \
&& pip cache remove \
onnxruntime-gpu \
&& pip install --break-system-packages \
&& pip --no-cache-dir install --break-system-packages \
onnxruntime-gpu \
--index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \
--extra-index-url https://pypi.org/simple \
Expand Down
4 changes: 1 addition & 3 deletions megapak/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -174,9 +174,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
-r /builder-scripts/pak7.txt \
&& pip uninstall --break-system-packages --yes \
onnxruntime-gpu \
&& pip cache remove \
onnxruntime-gpu \
&& pip install --break-system-packages \
&& pip --no-cache-dir install --break-system-packages \
onnxruntime-gpu \
--index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \
--extra-index-url https://pypi.org/simple \
Expand Down

0 comments on commit bdb9a45

Please sign in to comment.