tensorflow的gpu版本怎么安装

安装tensorflow的GPU版本，需要先安装CUDA和cuDNN。以下是安装步骤：

1. 下载和安装CUDA

在NVIDIA官网下载适合自己系统的CUDA安装包，然后按照安装向导进行安装。

2. 下载和安装cuDNN

在NVIDIA官网下载适合自己系统的cuDNN安装包，然后按照安装向导进行安装。

3. 创建conda虚拟环境

打开终端，输入以下命令创建一个名为“tensorflow-gpu”的conda虚拟环境：

conda create -n tensorflow-gpu python=3.6

4. 激活虚拟环境

输入以下命令激活“tensorflow-gpu”虚拟环境：

conda activate tensorflow-gpu

5. 安装tensorflow-gpu

输入以下命令安装tensorflow-gpu：

pip install tensorflow-gpu

6. 测试

输入以下代码测试是否安装成功：

import tensorflow as tf

tf.test.is_gpu_available() # 输出True表示安装成功

以上就是tensorflow-gpu版本的安装步骤。