lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update requirements for all cpu install (#666)

Browse Source 
Co-authored-by: Haifeng Jin <haifeng-jin@users.noreply.github.com>
This commit is contained in:
Haifeng Jin 2023-08-04 17:21:45 -07:00 committed by Francois Chollet
parent c58985b02d
commit c52df5381b
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements.txt
View File

@ -1,10 +1,9 @@
# Tensorflow # Tensorflow
tensorflow>=2.13.0 tensorflow-cpu>=2.13.0
# Torch. # Torch.
# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF
torch>=2.0.1+cpu torch>=2.0.1+cpu
torchvision>=0.15.1 torchvision>=0.15.1+cpu
# Jax. # Jax.
jax[cpu] jax[cpu]