Skip to content

Commit 752e72b

Browse files
committed
add update script
1 parent dd0274d commit 752e72b

File tree

1 file changed

+47
-0
lines changed

1 file changed

+47
-0
lines changed

update-pytorch.sh

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
#!/bin/bash
2+
3+
for d in . cpu cpu-cxx11-abi cpu_pypi_pkg cu75 cu80 cu90 cu91 cu92 cu100 cu101 cu102 cu110 cu111 cu113 cu115 cu116 cu117 cu117_pypi_cudnn cu118 cu121 cu124 rocm3.10 rocm3.7 rocm3.8 rocm4.0.1 rocm4.1 rocm4.2 rocm4.3.1 rocm4.5.2 rocm5.0 rocm5.1.1 rocm5.2 rocm5.3 rocm5.4.2 rocm5.5 rocm5.6 rocm5.7 rocm6.0 rocm6.1
4+
do
5+
d="whl/$d"
6+
[ "$d" == "whl/." ] && d="whl"
7+
mkdir -p $d/simple
8+
dir="$(pwd)"
9+
cd $d/simple
10+
curl -s https://download.pytorch.org/$d/ | grep -v 'TIMESTAMP 1' > index.html
11+
count="$(cat index.html | cut -d '>' -f 2 | cut -d '<' -f 1 | grep -cve '^\s*$')"
12+
13+
echo "https://download.pytorch.org/$d/ $count => $d/simple/"
14+
if [ $count -lt 40 ]
15+
then
16+
echo "failing because low packages count for $d: $count (intermittent download failure?)"
17+
exit 1
18+
fi
19+
20+
i=0
21+
for p in `cat index.html | cut -d '>' -f 2 | cut -d '<' -f 1`
22+
do
23+
mkdir $p
24+
cd $p
25+
((i++))
26+
curl -s https://download.pytorch.org/$d/$p/ \
27+
| sed -e 's_href="/whl_href="https://download.pytorch.org/whl_' \
28+
| grep -v 'TIMESTAMP 1' \
29+
> index.html
30+
31+
count="$(cat index.html | grep -c 'https://download.pytorch.org/whl/')"
32+
echo "$i $d/$p/ => $d/simple/$p/ $count"
33+
if [ $count -lt 1 ]
34+
then
35+
echo "failing because low packages count for $d/$p: $count (intermittent download failure?)"
36+
exit 1
37+
fi
38+
cd ..
39+
done
40+
echo
41+
cd "$dir"
42+
done
43+
44+
du -sh whl/*
45+
46+
for d in whl/simple whl/*/simple ; do echo "$(ls $d | wc -l) $d" ; done > summary.txt
47+
cat summary.txt

0 commit comments

Comments
 (0)