win10 + python3.6.1 + tensorflow1.10 + cuda9.0 + cudnn7.2

参考博客:https://blog.csdn.net/zw__chen/article/details/79374467、http://www.bubuko.com/infodetail-2730767.html

注意事项:
1、Tensorflow(CPU)与Tensorflow(GPU)不能共存,先装GPU再装CPU,也会被覆盖。最好就是删除CPU版本,只留下GPU版本
2、下载CUDA9.0.exe,下一步点到底
3、下载CuDnn,解压到C:\Program Files\cudnn-9.0-windows10-x64-v7.2.1.38,将文件中三个文件复制到cuda对应文件夹下。

4、把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64\cupti64_90.dll拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
4、CUDA的环境变量安装完成会配置一部分,最好再检查一下,总共bin,libnvvp,lib\x64,extras\CUPTI\libx64四个路径。
  cuDnn:将C:\Program Files\cudnn-9.0-windows10-x64-v7.2.1.38\cuda\bin 添加到path中
5、重启使环境变量生效
6、Test并观察是否用了GPU

猜你喜欢

转载自www.cnblogs.com/shiganquan/p/9496038.html