TensorFlow GPU配置

1.去TensorFlow官网查看配置信息

https://tensorflow.google.cn/install
只需要确定cuda版本和TF版本即可,因为cuda安装文件可以自动安装Nvidia显卡驱动

2.下载TF

直接安装推荐的最新版本即可
pip install tensorflow-gpu==2.0.0-beta1

3.下载cuda

直接百度cuda可以找到官网
注意官网默认下载是最新的版本,可能与TF查到的版本不一致,所以要选择历史版本(Legacy Release)

找到相对应的cuda版本和说明文档分别打开:

下载相应的cuda安装包

根据文档说明进行安装:
https://docs.nvidia.com/cuda/archive/10.0/

sudo service lightdm stop

4.安装cudnn

百度搜索cudnn找到官方的文档