Skip to content

Commit 01a813d

Browse files
authored
Merge pull request #17 from alimusdu/cuda-nightly
CUDA 12.9 and nightly added.
2 parents 79790bf + 709bb3d commit 01a813d

File tree

2 files changed

+15
-4
lines changed

2 files changed

+15
-4
lines changed

README.md

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ As a [Nexus Repository](https://help.sonatype.com/en/sonatype-nexus-repository.h
2424
- define name: `pypi-pytorch` for example
2525
- define URL for remote storage: `https://sonatype-nexus-community.github.io/pytorch-pypi/whl/`
2626

27-
You can also create compute platform specific proxies if necessary, like `pypi-pytorch-cu124` pointing to `https://sonatype-nexus-community.github.io/pytorch-pypi/whl/cu124/`
27+
You can also create compute platform specific proxies if necessary, like `pypi-pytorch-cu129` pointing to `https://sonatype-nexus-community.github.io/pytorch-pypi/whl/cu129/`
28+
29+
For nightly versions, simply create a `pypi-pytorch-nightly` repository pointing to `https://sonatype-nexus-community.github.io/pytorch-pypi/whl/nightly/`
2830

2931
These PyPI proxies can also be added to your `pypi-all` group repository if you created one.
3032

@@ -33,11 +35,19 @@ These PyPI proxies can also be added to your `pypi-all` group repository if you
3335
Newly created PyPI proxies can now be used with `pip` using provided `--index-url`, like:
3436

3537
```
36-
pip3 install pytorch+cu124 --index-url http://localhost:8081/repository/pypi-pytorch/simple
38+
pip3 install torch==2.8.0+cu129 --index-url http://localhost:8081/repository/pypi-pytorch/simple
3739
```
3840
or
3941
```
40-
pip3 install pytorch --index-url http://localhost:8081/repository/pypi-pytorch-cu124/simple
42+
pip3 install torch --index-url http://localhost:8081/repository/pypi-pytorch-cu129/simple
43+
```
44+
for nightly
45+
```
46+
pip3 install torch --index-url http://localhost:8081/repository/pypi-pytorch-nightly/simple
47+
48+
or
49+
50+
pip3 install torch==2.8.0.dev20250613+cu129 http://localhost:8081/repository/pypi-pytorch-nightly/simple
4151
```
4252

4353
## Development

update-pytorch.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@ EOF
102102

103103
# update main PyTorch index, that contains everything, whatever the compute platform
104104
updateIndex "whl"
105+
updateIndex "whl/nightly"
105106

106107
# see resulting updates
107108
git update-index -q --refresh
@@ -119,7 +120,7 @@ echo "updates found in PyTorch main index: updating also compute platform specif
119120
# ignore old non-updated ones:
120121
# - cu75 cu80 cu90 cu91 cu92 cu100 cu101 cu102 cu110 cu111 cu113 cu115 cu116 cu117 cu117_pypi_cudnn
121122
# - 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
122-
for d in cpu cpu-cxx11-abi cpu_pypi_pkg cu118 cu121 cu124 cu126 cu128 rocm6.0 rocm6.1 rocm6.2 rocm6.2.4 rocm6.3 xpu
123+
for d in cpu cpu-cxx11-abi cpu_pypi_pkg cu118 cu121 cu124 cu126 cu128 cu129 rocm6.0 rocm6.1 rocm6.2 rocm6.2.4 rocm6.3 rocm6.4 xpu
123124
do
124125
updateIndex "whl/$d"
125126
done

0 commit comments

Comments
 (0)