💻查看TensorFlow是GPU版还是CPU版🌟
发布时间：2025-04-01 02:38:12来源：
小伙伴们在使用TensorFlow时，是不是有时会好奇自己安装的是支持GPU加速的版本，还是只能靠CPU慢慢运算的普通版本呢？别担心，这里教大家如何快速检查！✨
首先，打开你的Python环境，输入以下代码👇：
```python
import tensorflow as tf
print("TensorFlow 版本:", tf.__version__)
print("是否支持GPU:", tf.test.is_built_with_gpu_support())
print("是否有可用的GPU设备:", tf.config.list_physical_devices('GPU'))
```
这段代码会告诉你TensorFlow的版本号，以及当前系统是否支持GPU和是否有实际的GPU设备被识别。如果`is_built_with_gpu_support()`返回True，说明你安装的是支持GPU的版本；而`list_physical_devices('GPU')`会列出所有可用的GPU设备。如果有设备名称显示，那就说明你的环境已经成功连接到GPU啦！🎉
记得检查一下显卡驱动是否正确安装哦，这样才能让GPU火力全开，加速深度学习任务！🔥
免责声明：本答案或内容为用户上传，不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。 如遇侵权请及时联系本站删除。
