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
找到官方的文档