Skip to content

Commit c6d2283

Browse files
committed
Move no-install-package to docker container
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent e3c4b58 commit c6d2283

2 files changed

Lines changed: 17 additions & 19 deletions

File tree

docker/Dockerfile.ci

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,5 +31,21 @@ RUN --mount=type=bind,source=pyproject.toml,target=/workspace/pyproject.toml \
3131

3232
# Use the container's torch installation rather than reinstall it
3333
uv venv --system-site-packages ${UV_PROJECT_ENVIRONMENT}
34-
uv sync --link-mode symlink --locked --all-groups
34+
uv sync --link-mode symlink --locked --all-groups \
35+
--no-install-package absl-py \
36+
--no-install-package torch \
37+
--no-install-package torchvision \
38+
--no-install-package triton \
39+
--no-install-package nvidia-cublas-cu12 \
40+
--no-install-package nvidia-cuda-cupti-cu12 \
41+
--no-install-package nvidia-cuda-nvrtc-cu12 \
42+
--no-install-package nvidia-cuda-runtime-cu12 \
43+
--no-install-package nvidia-cudnn-cu12 \
44+
--no-install-package nvidia-cufft-cu12 \
45+
--no-install-package nvidia-cufile-cu12 \
46+
--no-install-package nvidia-curand-cu12 \
47+
--no-install-package nvidia-cusolver-cu12 \
48+
--no-install-package nvidia-cusparse-cu12 \
49+
--no-install-package nvidia-cusparselt-cu12 \
50+
--no-install-package nvidia-nccl-cu12
3551
EOF

pyproject.toml

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -89,24 +89,6 @@ dev = [
8989

9090
[tool.uv]
9191
managed = true
92-
override-dependencies = [
93-
"absl-py ; sys_platform == 'never'",
94-
"torch ; sys_platform == 'never'",
95-
"torchvision ; sys_platform == 'never'",
96-
"triton ; sys_platform == 'never'",
97-
"nvidia-cublas-cu12 ; sys_platform == 'never'",
98-
"nvidia-cuda-cupti-cu12 ; sys_platform == 'never'",
99-
"nvidia-cuda-nvrtc-cu12 ; sys_platform == 'never'",
100-
"nvidia-cuda-runtime-cu12 ; sys_platform == 'never'",
101-
"nvidia-cudnn-cu12 ; sys_platform == 'never'",
102-
"nvidia-cufft-cu12 ; sys_platform == 'never'",
103-
"nvidia-cufile-cu12 ; sys_platform == 'never'",
104-
"nvidia-curand-cu12 ; sys_platform == 'never'",
105-
"nvidia-cusolver-cu12 ; sys_platform == 'never'",
106-
"nvidia-cusparse-cu12 ; sys_platform == 'never'",
107-
"nvidia-cusparselt-cu12 ; sys_platform == 'never'",
108-
"nvidia-nccl-cu12 ; sys_platform == 'never'"
109-
]
11092

11193
[tool.setuptools]
11294
packages = ["emerging_optimizers"]

0 commit comments

Comments
 (0)