win10安装tensorflow-gpu

  1. 安装cuda9.2。网址https://developer.nvidia.com/cuda-toolkit-archive
  2. 查看cuda版本:nvcc --version
  3. 安装cudnn。下载网址https://developer.nvidia.com/rdp/cudnn-archive。安装方法:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-windows
  4. 安装anaconda。
  5. 新建虚拟环境,在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

发表评论

邮箱地址不会被公开。 必填项已用*标注