File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 5454third_party_path = os .path .join (base_dir , 'third_party' )
5555
5656_libtpu_version = '0.1.dev20220518'
57- _litbpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{ _libtpu_version } -py3-none-any.whl'
57+ _libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{ _libtpu_version } -py3-none-any.whl'
5858
5959
6060def _get_build_mode ():
@@ -354,10 +354,14 @@ def make_relative_rpath(path):
354354 [make_relative_rpath ('torch_xla/lib' )],
355355 ),
356356 ],
357+ install_requires = [
358+ 'absl-py>=1.0.0' ,
359+ 'cloud-tpu-client>=0.10.0' ,
360+ ],
357361 extras_require = {
358362 # On Cloud TPU VM install with:
359363 # $ sudo pip3 install torch_xla[tpuvm] -f https://storage.googleapis.com/tpu-pytorch/wheels/tpuvm/torch_xla-1.11-cp38-cp38-linux_x86_64.whl
360- 'tpuvm' : [f'libtpu-nightly @ { _litbpu_storage_path } ' ],
364+ 'tpuvm' : [f'libtpu-nightly @ { _libtpu_storage_path } ' ],
361365 },
362366 package_data = {
363367 'torch_xla' : [
You can’t perform that action at this time.
0 commit comments