wehub-resource-sync
13 lines
268 B
Plaintext
13 lines
268 B
Plaintext
# Tensorflow with cuda support.
|
|
tensorflow[and-cuda]==2.20.0
|
|
ai-edge-litert
|
|
|
|
# Torch cpu-only version (needed for testing).
|
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
torch
|
|
|
|
# Jax cpu-only version (needed for testing).
|
|
jax[cpu]
|
|
|
|
-r requirements-common.txt
|