@@ -166,8 +166,7 @@ wheels for `torch`, `torchvision`, and `torch_xla` at
166166| 2.0 + CUDA 11.8 | ` https://storage.googleapis.com/tpu-pytorch/wheels/cuda/118/torch_xla-2.0-cp38-cp38-linux_x86_64.whl ` |
167167| 2.0 + CUDA 11.7 | ` https://storage.googleapis.com/tpu-pytorch/wheels/cuda/117/torch_xla-2.0-cp38-cp38-linux_x86_64.whl ` |
168168| 1.13 | ` https://storage.googleapis.com/tpu-pytorch/wheels/cuda/112/torch_xla-1.13-cp38-cp38-linux_x86_64.whl ` |
169- | nightly + CUDA 11.7 <= 2023/04/25| ` https://storage.googleapis.com/tpu-pytorch/wheels/cuda/117/torch_xla-nightly-cp38-cp38-linux_x86_64.whl ` |
170- | nightly + CUDA 11.7 >= 2023/04/25| ` https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/11.7/torch_xla-nightly-cp38-cp38-linux_x86_64.whl ` |
169+ | nightly + CUDA 12.0 >= 2023/06/27| ` https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/12.0/torch_xla-nightly-cp38-cp38-linux_x86_64.whl ` |
171170| nightly + CUDA 11.8 <= 2023/04/25| ` https://storage.googleapis.com/tpu-pytorch/wheels/cuda/118/torch_xla-nightly-cp38-cp38-linux_x86_64.whl ` |
172171| nightly + CUDA 11.8 >= 2023/04/25| ` https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/11.8/torch_xla-nightly-cp38-cp38-linux_x86_64.whl ` |
173172
@@ -215,6 +214,13 @@ nightly at date(< 2023/04/25) | `gcr.io/tpu-pytorch/xla:nightly_3.8_tpuvm_YYYYMM
215214
216215<br />
217216
217+ | Version | GPU CUDA 12.0 + Python 3.8 Docker |
218+ | --- | ----------- |
219+ | nightly | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.8_cuda_12.0 ` |
220+ | nightly at date(>=2023/06/27) | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.8_cuda_12.0_YYYYMMDD ` |
221+
222+ <br />
223+
218224| Version | GPU CUDA 11.8 + Python 3.8 Docker |
219225| --- | ----------- |
220226| 2.0 | ` gcr.io/tpu-pytorch/xla:r2.0_3.8_cuda_11.8 ` |
@@ -227,9 +233,6 @@ nightly at date(< 2023/04/25) | `gcr.io/tpu-pytorch/xla:nightly_3.8_tpuvm_YYYYMM
227233| Version | GPU CUDA 11.7 + Python 3.8 Docker |
228234| --- | ----------- |
229235| 2.0 | ` gcr.io/tpu-pytorch/xla:r2.0_3.8_cuda_11.7 ` |
230- | nightly | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.8_cuda_11.7 ` |
231- | nightly at date(>=2023/04/25) | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.8_cuda_11.7_YYYYMMDD ` |
232- | nightly at date(<2023/04/25) | ` gcr.io/tpu-pytorch/xla:nightly_3.8_cuda_11.7_YYYYMMDD ` |
233236
234237<br />
235238
0 commit comments