tensorflow使用gpu 但是运行还是大量使用cpu训练速度也没有提升

2025-03-20 13:47:02
推荐回答(1个)
回答1:

安装GPU版本的tensorflow就可以了pipinstall--upgrade还需要安装GPU加速包,下载CUDA和cuDNN的安装包了,注意版本号分别是CUDA8.0和cuDNN5.1。
如果都安装成功,运行程序时会自动使用GPU计算