如何安装tensorflow gpu 1.15.4

以下是安装TensorFlow GPU 1.15.4的步骤：

1. 确保您的计算机已安装了支持CUDA 10.0的GPU驱动程序。您还需要安装CUDA Toolkit 10.0和cuDNN v7.4.2（TensorFlow GPU 1.15.4支持的版本）。

2. 创建一个虚拟环境（可选）。这可以帮助您避免与其他Python库的冲突。

3. 使用pip安装TensorFlow GPU 1.15.4：

pip install tensorflow-gpu==1.15.4

4. 如果您遇到了任何问题，请确保您的环境变量正确设置。您需要将CUDA和cuDNN添加到PATH和LD_LIBRARY_PATH中。您可以通过以下命令检查这些环境变量：

echo $PATH
echo $LD_LIBRARY_PATH

如果环境变量未正确设置，请按照CUDA和cuDNN的安装文档进行操作。

5. 您现在已经成功安装了TensorFlow GPU 1.15.4！您可以通过以下方式在Python中验证它是否正常工作：

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

如果您的GPU已正确识别并分配给TensorFlow，您将看到类似以下内容的输出：

Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce GTX 1080, pci bus id: 0000:01:00.0, compute capability: 6.1

祝您好运！