环境
- windows 10 64bit
- GPU GeForce GTX 1060 6GB
不使用Anaconda安装
python
- tensorflow在Windows还未支持python3.6,需要安装python3.5
- 下载地址python3.5.4
Visual Studio 2015
- 安装cuda需要使用VC++编译,所以需要安装VS,安装VS时可以只选择需要的Visual C++
- 安装的是社区版update3
- 地址 visual studio 2015
- 安装选项中只选择了Visual C++
CUDA 8.0
- 下载地址 cuda8.0
- 安装CUDA时需要关闭VS2015,并选择自定义安装,取消勾选driver,可以安装成功
- 安装完成后,在cmd中输入 nvcc -V 可以查看安装的CUDA版本,测试是否安装成功
- 安装完成后,打开VS2015,新建NVIDIA项目,直接回建一个示例程序,可以跑通示例就ok
cuDNN6.0
- 需要注册方可下载下载地址
- 直接下载地址:cudnn6.0 for Cuda8.0
- 直接下载地址:cudnn5.1 for Cuda8.0
- 本文安装的是cudnn6.0版本,下载完成后解压压缩包,将对应目录下的文件移到相应目录
- e.g. \cudnn-8.0-windows10-x64-v6.0\cuda\bin 下的cudnn64_6.dll复制到 \Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 下
安装Tensorflow
- pip3 install tensorflow-gpu
- 可能会出现如下错误{ PermissionError: [WinError 5] 拒绝访问。: ‘c:\program files\python35\Lib\site-packages\enum’ }
- 某些文件不具有访问权限,可以使用管理员权限打开cmd,再安装,也可以修改这些文件的权限
- 附权限问题(可以尝试,本文直接省事管理员权限了)
测试
- 在cmd中输入python,进入python3.5
- 输入 :import tensorflow as tf
- 没有报错,就表示安装成功
安装中出现的问题
- 安装CUDA时,电脑出现闪屏或者黑屏,(最后一次成功安装未出现长时间黑屏–超过30s,之前失败安装出现长时间黑屏10min+,最后就直接手动重启了)
- Tensorflow网站上可以找到支持python3.6的1.4版本,但是之前为安装成功,转战到python3.5了,想尝试的同鞋可以试一下