之前一直用的tensorflow cpu版的,后来数据量大一些之后,觉得有点算不动了,所以打算改用gpu版。
传闻gpu不太容易安装,谁知道这么坑!
光系统就重装了3遍,当然这里面是我自己有点强迫症,最后感觉能否安装成功与系统关系不大。下面是正经内容。
系统:cn_windows_7_ultimate_with_sp1_x64_dvd_u_677408(这个不重要)
笔记本配置:处理器Intel(R) Core(TM) i7-4710MQ CPU @2.50GHz,内存16G,显卡NVIDIA GeForce GTX 860M
python版本:3.5.2
tensorflow版本:tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl
安装过程:
1. 先安装Python3.5.2,记得添加路径,这个大家应该很熟了,不做过多介绍。
2. 安装Visual C++ Redistributable for Visual Studio 2015(https://www.microsoft.com/en-us/download/details.aspx?id=48145),或者直接安装Visual Studio 2015。
3.安装CUDA和cudnn
CUDA版本:cuda v8.0.60(cuda_8.0.61_windows.exe和cuda_8.0.61.2_windows.exe,注意