gpt4 book ai didi

machine-learning - ML 训练过程不在 GPU 上

转载 作者:行者123 更新时间:2023-11-30 08:46:09 27 4
gpt4 key购买 nike

我刚刚从 AWS 迁移到 Google Cloud Platform,因为它的 GPU 价格较低。我按照网站上的说明创建了一个带有 K80 GPU 的计算引擎实例,安装了最新版本的 Tensorflow、keras、cuda 驱动程序和 cnDNN,一切都很顺利。然而,当我尝试训练我的模型时,训练过程仍在 CPU 上进行。

NVIDIA-SMI 387.26                 Driver Version: 387.26        
Cuda compilation tools, release 9.1, V9.1.85
Tensorflow version -1.4.1
cudnn cudnn-9.1-linux-x64-v7
Ubuntu 16.04

最佳答案

也许你安装了tensorflow的cpu版本?

现在,Google Cloud 的计算引擎拥有一个虚拟机操作系统镜像,其中安装了所有所需的软件,可以更轻松/更快地开始使用:https://cloud.google.com/deep-learning-vm/

关于machine-learning - ML 训练过程不在 GPU 上,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47951209/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com