gpt4 book ai didi

tensorflow - 无法在 Ubuntu 16.04 中配置 tensorflow 以使用 GPU 加速

转载 作者:行者123 更新时间:2023-12-04 19:09:44 25 4
gpt4 key购买 nike

我正在尝试在 Ubuntu 16.04(在谷歌云中)安装 Tensorflow。到目前为止,我所做的是创建了一个计算实例。我在这个实例中添加了一个 NVIDIA Tesla K80。

enter image description here

此外,请确保安装了正确版本的 tensorflow(版本 1.14.0)并

安装了cuda 8.0版本

根据 tensorflow gpu - cuda mapping 安装 CudNN 6.0 版本

当我运行一个简单的 tensorflow 程序时,我得到

无法为操作 MatMul 分配设备:{{node MatMul}} 已明确分配给/device:GPU:0 但可用设备为 [/job:localhost/replica:0/task:0/device:CPU:0,/job :localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ]。 确保设备规范引用有效设备 .

谁能让我知道我在哪里做错了。实例选择是否正确?

请让我知道并感谢您的帮助。

最佳答案

使用 Tensorflow 1.14 测试的 CUDA 和 CudNN 版本分别是 10.0 和 7.4。

有关版本兼容性的更多信息,请访问 here .

关于tensorflow - 无法在 Ubuntu 16.04 中配置 tensorflow 以使用 GPU 加速,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59596553/

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