requirements-tpu.txt 228 Bytes
Newer Older
1
2
3
4
5
6
7
# Common dependencies
-r requirements-common.txt

# Dependencies for TPU
# Currently, the TPU backend uses a nightly version of PyTorch XLA.
# You can install the dependencies in Dockerfile.tpu.
triton  # To avoid import errors