update requirements for all cpu install (#666)
Co-authored-by: Haifeng Jin <haifeng-jin@users.noreply.github.com>
This commit is contained in:
parent
c58985b02d
commit
c52df5381b
@ -1,10 +1,9 @@
|
||||
# Tensorflow
|
||||
tensorflow>=2.13.0
|
||||
tensorflow-cpu>=2.13.0
|
||||
|
||||
# Torch.
|
||||
# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF
|
||||
torch>=2.0.1+cpu
|
||||
torchvision>=0.15.1
|
||||
torchvision>=0.15.1+cpu
|
||||
|
||||
# Jax.
|
||||
jax[cpu]
|
||||
|
Loading…
Reference in New Issue
Block a user