Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
lanluo-nvidia committed Aug 9, 2024
1 parent adaf1e8 commit 0e2f646
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 3 deletions.
2 changes: 2 additions & 0 deletions .github/scripts/disk_monitor.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,7 @@ do
docker system df
echo "df -kh ."
df -kh .
echo "du -h --max-depth=1 ."
du -h --max-depth=1 .
sleep 1m
done
5 changes: 3 additions & 2 deletions .github/workflows/docker_builder.yml
Original file line number Diff line number Diff line change
Expand Up @@ -54,10 +54,11 @@ jobs:
python3 -m pip install pyyaml
TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
echo "TRT VERSION = ${TRT_VERSION}"
du -h --max-depth=1 .
df -kh .
docker system df
docker container prune -f
docker image prune -f
docker system prune -f
docker system df
.github/scripts/disk_monitor.sh &
DOCKER_BUILDKIT=1 docker build --build-arg TENSORRT_VERSION=$TRT_VERSION -f docker/Dockerfile --tag $DOCKER_TAG .
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
torchvision>=0.19.0.dev,<0.20.0
--extra-index-url https://pypi.ngc.nvidia.com
pyyaml
Expand Down

0 comments on commit 0e2f646

Please sign in to comment.