Windows10/xxx显卡安装TensorFlow-GPU
安装显卡驱动
1、桌面右键点击打开NIVDIA控制面板。
2、点开帮助里面的系统信息。
3、查看本子自带驱动版本。
4、查看支持的CUDA版本。
安装CUDA
1、从上面第二个图看出本机显卡支持CUDA-9.2。
2、从官网下载安装程序cuda_9.2.148_win10.exe。
3、运行cuda_9.2.148_win10.exe,安装选项默认是精简,选择自定义。点下一步,在自定义选项中去掉了 Visual Studio Integration。
安装CUDNN
1、从官网下载与CUDA9.2对应的cudnn7.2压缩文件cudnn-9.2-windows10-x64-v7.2.1.38.zip。