Install `tensorflow-cpu` in CIs

optuna / optuna-examples

Examples for https://github.com/optuna/optuna

MIT License

632 stars 171 forks source link

Install `tensorflow-cpu` in CIs #260

Closed not522 closed 1 month ago

not522 commented 2 months ago

Motivation

In the CPU-only environment, tensorflow-cpu is recommended. https://www.tensorflow.org/install/pip It has a smaller package size and may reduce errors during download in the CIs. For example, the size of tensorflow-2.16.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl is 589.8 MB, but tensorflow_cpu-2.16.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl is 214.1 MB.

Description of the changes

Install tensorflow-cpu instead of tensorflow.

not522 commented 2 months ago

@nabenabe0928 Could you review this PR?

github-actions[bot] commented 1 month ago

This pull request has not seen any recent activity.