- 安装cuda9.2。网址https://developer.nvidia.com/cuda-toolkit-archive。
- 查看cuda版本:
nvcc --version
- 安装cudnn。下载网址https://developer.nvidia.com/rdp/cudnn-archive。安装方法:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-windows。
- 安装anaconda。
- 新建虚拟环境,在https://github.com/fo40225/tensorflow-windows-wheel下载对应版本的tensorflow-gpu。
官网推出的tensorflow-gpu支持到9.0,至于为什么安装cuda9.2,是因为安装cuda9.0会出问题https://stackoverflow.com/questions/53753676/blas-gemv-launch-failed-m-3-n-2-opmatmul/53783188#53783188。