GTX1050Ti+CUDN10.0+cudnn7.6.0+tensorflow-gpu1.18.0
win10+pycharm+annaconda搭建tensorflow-gpu
1.检查电脑显卡
2.去对应的显卡官网检查是否支持开启GPU
3.查找NVCUDA.dll的版本,看看当前显卡驱动支持什么版本的CUDA(即安装cuda版本不能高于当前显示的.dll版本,如果可支持的版本太低,那就更新驱动,这样可以适配更高的cuda)
4.下载可支持版本的cuda,安装
5.下载当前版本cuda对应的cudnn,解压得到的文件分别并入cuda(我理解这是一个cuda的扩展包,用与支持开启cuda)
6.下载对应版本的tensorflow-gpu
貌似是:
tf1.13以下 cuda9.0
tf1.14-2.0.0 cuda10.0
tf2.1.0以上 cuda10.1
7.ok