Skip to content

Commit b920c7a

Browse files
committed
update only recent compute platforms
1 parent 752e72b commit b920c7a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

update-pytorch.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
#!/bin/bash
22

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
3+
# old non-updated compute platforms:
4+
# cu75 cu80 cu90 cu91 cu92 cu100 cu101 cu102 cu110 cu111 cu113 cu115 cu116 cu117 cu117_pypi_cudnn
5+
# 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
6+
for d in . cpu cpu-cxx11-abi cpu_pypi_pkg cu118 cu121 cu124 rocm6.0 rocm6.1
47
do
58
d="whl/$d"
69
[ "$d" == "whl/." ] && d="whl"

0 commit comments

Comments
 (0)