Skip to content

Commit

Permalink
build: unpin tensorflow version on windows (deepmodeling#3721)
Browse files Browse the repository at this point in the history
tensorflow/tensorflow#61830 got fixed.

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **Bug Fixes**
- Updated the TensorFlow version constraint for better compatibility on
Windows systems.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
  • Loading branch information
njzjz authored and mtaillefumier committed Sep 18, 2024
1 parent cc39eab commit c367e54
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion backend/find_tensorflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def get_tf_requirement(tf_version: str = "") -> dict:
"tensorflow-cpu; platform_machine!='aarch64' and (platform_machine!='arm64' or platform_system != 'Darwin')",
"tensorflow; platform_machine=='aarch64' or (platform_machine=='arm64' and platform_system == 'Darwin')",
# https://github.com/tensorflow/tensorflow/issues/61830
"tensorflow-cpu<2.15; platform_system=='Windows'",
"tensorflow-cpu!=2.15.*; platform_system=='Windows'",
*extra_requires,
],
"gpu": [
Expand Down

0 comments on commit c367e54

Please sign in to comment.