使用 GPU

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

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