首页
科技
登录
6mi
u
盘
搜
搜 索
科技
检查安装的tensorflow版本信息
检查安装的tensorflow版本信息
科技
2024-03-24
78
1、tensorflow信息
tf.__version__ # 版本信息 tf.__path__ # 安装位置
2、是否是GPU
tf.test.is_gpu_available() # 输出为True则为GPU
3、选择GPU还是CPU
整体来说,代码在这两种环境下运行几乎没什么区别,但GPU更加适合复杂密集的高速运算,像入手深度学习进行模型训练首选GPU。
转载请注明原文地址:https://blackberry.8miu.com/read-30046.html
最新回复
(
0
)