---
title: 安装TensorFlow
description: "1. 使用PuTTY工具，以root用户登录服务器。"
url: https://www.hikunpeng.com/document/detail/zh/kunpenghpcs/hpcindapp/prtg-osc/kunpengalphafold_02_0008.html
sourcePath: /source/zh/kunpenghpcs/hpcindapp/prtg-osc/kunpengalphafold_02_0008.html
indexId: 59b8d37f2c0fda31b9ce6ba7ac87f4e7deddcaa5b9585f5aa75451bfe83743ec71
---
# 安装TensorFlow

#### 操作步骤

1. 使用PuTTY工具，以root用户登录服务器。
2. 执行以下命令下载TenforFlow源码。

```
git clone https://github.com/tensorflow/tensorflow.git
```


3. GPU版本需要安装cuda-toolkit（请自行安装）。
4. 执行以下命令配置TensorFlow编译选项。

```
cd tensorflows-xxx
./configure
```


  GPU版本配置如下：

5. 执行以下命令编译生成.whl文件。

```
bazel clean –expunge
```


6. GPU版本编译命令。

```
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
mkdir tensorflow-pkg
bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu ./tensorflow-pkg
```


7. 生成的.whl文件在tensorflow_pkg目录下，执行以下命令安装。

```
pip install tensorflow-xxx.whl
```