Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix docker build issue #3070

Merged
merged 15 commits into from
Aug 16, 2024
2 changes: 1 addition & 1 deletion docker/WORKSPACE.docker
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ load("@rules_python//python:pip.bzl", "pip_parse")

pip_parse(
name = "devtools_deps",
requirements = "//:requirements-dev.txt",
requirements_lock = "//:requirements-dev.txt",
lanluo-nvidia marked this conversation as resolved.
Show resolved Hide resolved
)

load("@devtools_deps//:requirements.bzl", "install_deps")
Expand Down
18 changes: 17 additions & 1 deletion py/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,23 @@ numpy
packaging
pybind11==2.6.2
--extra-index-url https://download.pytorch.org/whl/nightly/cu124
torch>=2.5.0.dev,<2.6.0
# once we are able to bump torchvision to 0.20.0, change it back
# it is trying to download all the torch versions from 0808 back to 0617 to find the matching torchvision 0.19.0 which cause disk space to be full
# #30 43.91 Collecting torch<2.6.0,>=2.5.0.dev
#30 43.95 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240808%2Bcu124-cp310-cp310-linux_x86_64.whl (992.9 MB)
#30 53.10 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 992.9/992.9 MB 3.8 MB/s eta 0:00:00
#30 53.70 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240807%2Bcu124-cp310-cp310-linux_x86_64.whl (992.8 MB)
#30 62.36 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 992.8/992.8 MB 3.9 MB/s eta 0:00:00
#30 63.45 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240806%2Bcu124-cp310-cp310-linux_x86_64.whl (991.8 MB)
#30 107.3 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 991.8/991.8 MB 3.4 MB/s eta 0:00:00
#30 107.8 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240805%2Bcu124-cp310-cp310-linux_x86_64.whl (992.7 MB)
#30 114.6 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 992.7/992.7 MB 3.5 MB/s eta 0:00:00
#30 115.2 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240804%2Bcu124-cp310-cp310-linux_x86_64.whl (992.7 MB)
#30 121.9 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 992.7/992.7 MB 3.9 MB/s eta 0:00:00
#30 122.4 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240803%2Bcu124-cp310-cp310-linux_x86_64.whl (992.7 MB)
#30 129.2 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 992.7/992.7 MB 3.9 MB/s eta 0:00:00
#30 129.8 Downloading https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240802%2Bcu124-cp310-cp310-linux_x86_64.whl (992.7 MB)
torch>=2.5.0.dev20240617,<2.5.0.dev20240618
lanluo-nvidia marked this conversation as resolved.
Show resolved Hide resolved
torchvision>=0.19.0.dev,<0.20.0
lanluo-nvidia marked this conversation as resolved.
Show resolved Hide resolved
--extra-index-url https://pypi.ngc.nvidia.com
pyyaml
Expand Down
Loading