【安装教程】安装tensorflow-gpu版本_tensorflowgpu版本

NOTE:

如果不是首次安装tensorflow，卸载TensorFlow相关的内容，包括依赖的包（tensorflow-estimator、tensorboard、tensorflow、keras-applications、keras-preprocessing），不然后续安装了tensorflow-gpu可能会出现找不到cuda的问题。
== 查看和卸载指令 ==

pip list #查看TensorFlow相关的包
pip uninstall tesnsorflow  #卸载和TensorFlow相关的包
...
1
2
3

准备工作

在安装TensorFlow之前，先装好cuda和anaconda
安装教程如下：
anaconda安装：anaconda安装
cuda安装：cuda11.6安装

TensorFlow安装

1、 确定TensorFlow版本

根据安装的CUDA版本、python版本确定对应的tensoflow-gpu版本、详细信息查看该网址TensorFlow安装网址

2、使用pip直接安装

打开cmd命令行界面，这里建议使用管理员身份打开。

或者在pycharm的终端中进行（我采用的方式）

3、输入安装指令

pip install tensorflow-gpu==2.6.0 --default-timeout=100 -i https://pypi.tuna.tsinghua.edu.cn/simple
1

安装成功！

TensorFlow安装后测试

TensorFlow安装后，进行测试是否可以正常运行，新建一个python文件，输入以下代码并运行

import tensorflow as tf
tf.autograph.set_verbosity(0)
physical_devices = tf.config.experimental.list_physical_devices('GPU')
print(physical_devices)
config = tf.config.experimental.set_memory_growth(physical_devices[0], True)

1
2
3
4
5
6

到此，TensorFlow—gpu版本安装成功！