如何在tensorflow中重新安装gpu答案

【问题标题】：How to reinstall gpu in tensorflow如何在tensorflow中重新安装gpu
【发布时间】：2019-07-31 12:14:57
【问题描述】：

我只安装了 tensorflow cpu，现在我想启用 gpu 功能。系统： Ubuntu 18.04.1 LTS 使用蟒蛇的python 3.7 张量流 1.13.1

【问题讨论】：

【解决方案1】：

查看tensorflow安装指南https://www.tensorflow.org/install/pip

【讨论】：

【解决方案2】：

Tensorflow-GPU 是附加包，不同于 Tensorflow CPU。

您需要兼容 CUDA 的 GPU 才能使用 tensorflow GPU！ 在开始之前，您需要安装：CUDNN、CUDA Toolkit、libcupti。

然后，使用 pip：

pip install --upgrade tensorflow-gpu

【讨论】：

它成功运行并全部安装.. 但是现在当我尝试使用带有 tensorflow 代码的 GPU 时（使用 tf.device('/gpu:0') ）我收到此错误：{InvalidArgumentError: Cannot为操作 random_normal_1/RandomStandardNormal 分配一个设备：节点 random_normal_1/RandomStandardNormal（在 /home/bignrz/untitled0.py:4 定义）被显式分配给 /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]。确保设备规范引用了有效的设备。}