windows10下安装tensorflow_gpu总结
我的win10 + gtx960m显卡 + tensorflow-gpu1.7-cp36 + CUDA9.0 + cuDNN7.0.4 + anaconda3.5.2即python3.6 完美安装
一.准备工作:
- 1)查看自己显卡支持的驱动CUDA
查看自己显卡能用的驱动
2)如何查看显卡支持的CUDA版本
3)CUDA官方文档
2) 查看anaconda与Python版本对应 - 查看tensorflow版本、python版本、编译器版本、cuDNN、CUDA对应关系
- 根据自己的电脑选择安装包下载,一般选择编译好的**.whl**文件,这样不用编译
tensorflow-gpu各版本whl下载地址
cuDNN各版本下载地址
CUDA各版本下载地址
anaconda各版本下载地址 :官网或者 清华镜像
anacondaminiconda镜像下载地址 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=N&O=D https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/?C=N&O=D官方首页https://www.anaconda.com/distribution/https://docs.conda.io/en/latest/miniconda.html
官方下载地址https://repo.anaconda.com/archive/https://repo.anaconda.com/miniconda/
官方文档https://docs.anaconda.com/anaconda/
old package lists
release notes
https://docs.anaconda.com/anaconda/reference/release-notes/二.开始安装
- 首先安装anaconda
- 安装CUDA,注意:
- 安装时关闭杀毒软件或者一直点击允许
- 安装中勾选添加环境变量,不然就得自己加
- 安装cuDNN,直接将cudnn解压得到的三个文件夹复制到CUDA安装目录下。
- 安装tensorflow-gpu,安装所用命令:pip install 文件路径
安装keras
tensorflow和keras版本对应关系 :博客
三.后期问题
- T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:396] Loaded runtime CuDNN library: 7500 (compatibility version 7500) but source was compiled with 7003 (compatibility version 7000). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
字面意思cuDNN7.5不兼容,需要换成cuDNN7.0.03
- 解决步骤:
- 1, 卸载原来的cuDNN,它只有三个文件夹且文件夹最里面只有3各文件,进入CUDA安装目录例如:v9.0下分别将这三个文件删除。
- 2, 将新下载的cuDNN解压得到的三个文件夹,再复制到v9.0目录下
- ModuleNotFoundError: No module named ‘_pywrap_tensorflow_internal’
Failed to load the native TensorFlow runtime.
- 解决步骤:
- 原因:版本不兼容,降低tensorflow版本
- 博客