1、Cuda
查看Cuda支持的GPU型号:https://developer.nvidia.com/cuda-gpus
下载cuda:https://developer.nvidia.com/cuda-downloads
下载好之后推荐安装就好了。
一般默认安装在 C:Program FilesNVIDIA GPU Computing ToolkitCUDA
将Cuda安装目录下的bin文件夹和libx64文件夹添加到path。
2、Cudnn
根据Cuda版本下载对应Cudnn:https://developer.nvidia.com/rdp/cudnn-archive
下载好后解压如图:
将上面三个文件夹里的东西分别放到cuda安装目录同名文件夹下面
3、tensorflow
pip uninstall tensorflow
pip install tensorflow-gpu==1.11.0
注意:使用pip是不能同时打开jupyter notebook,否则会失败。