tensorflow-gpu安装及注意事项

tensorflow-gpu安装

1.CUDA安装

打开显卡驱动，找到显卡对应的CUDA版本（如图所示 此电脑显卡对应的cuda为11.3.1）
帮助-》系统信息-》组件

1.1下载CUDA（https://developer.nvidia.com/cuda-toolkit-archive）

进入CUDA官网，选择对应的CUDA版本

解压，安装，安装时可自定义路径。

1.2CUDA配置环境变量

输入nvcc -V 出现如图信息，安装成功

2.安装cudnn（https://developer.nvidia.com/rdp/cudnn-archive）

下载cudnn之前需要注册账号，按照要求填信息即可，注册成功后即可下载。
选择CUDA版本对应的cudnn版本

将下载好的文件，解压。将里面三个文件夹（bin、include、lib）下的文件，复制到CUDA安装路径下的的同名文件中。

3.tensorflow-gpu安装

3.1查看CUDA版本对应的tensorflow版本（这里没有显示11.3.1对应的tensorflow版本，我这里选择安装2.6的版本也是可以的）（https://www.tensorflow.org/install/source_windows）

使用anaconda安装对应的tensorflow-gpu版本

3.2测试是否安装成功

import tensorflow as tf
tf.config.list_physical_devices('GPU')

4.参考链接

https://blog.csdn.net/sinat_23619409/article/details/84202651

https://blog.csdn.net/sunmingyang1987/article/details/102872658