GTX 960M+CUDA9.0+CUDNN 7.6.4+Python 3.6.5+TensorFlow-gpu 1.12.0+numpy 1.16.4
以上配置是我参考官方文档之后多次搭配之后所得到的组合,在配置过程中有几个注意点。有一样的小伙伴可以参考。
- 首先确定自己的显卡型号
- 安装CUDA一直下一步即可
- 将下载好的CUDNN里面的三个文件夹copy并替换到CUDA文件夹下
- 配置坏境变量
- 配置完之后一定要重启
- 测试是否安装成功,可能会出现numpy报错问题,升级或降级原有的numpy版本即可