使用 GPU

中文版 tensorflow
1

https://zh.d2l.ai/chapter_deep-learning-computation/use-gpu.html

2

经我测试，tensorflow2.4.0中一个张量在cpu上（默认），另一个在gpu上，它们可以直接相加，结果存在gpu上；而对于神经网络，默认直接创建在gpu上。
我看网上说，更高版本的tensorflow会默认把张量和网络都放在gpu上