We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
tpuvm
1 parent dc0acb4 commit 813ffe5Copy full SHA for 813ffe5
setup.py
@@ -313,8 +313,6 @@ def run(self):
313
# pip install torch_xla[tpu] -f https://storage.googleapis.com/libtpu-releases/index.html
314
'tpu': [f'libtpu-nightly=={_libtpu_version}'],
315
# On nightly, install libtpu with `pip install torch_xla[tpuvm]`
316
- # Remove from release branches since this is not allowed by PyPI.
317
- 'tpuvm': [f'libtpu-nightly @ {_libtpu_storage_path}'],
318
# pip install torch_xla[pallas] -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html
319
'pallas': [f'jaxlib=={_jax_version}', f'jax=={_jax_version}'],
320
},
0 commit comments