windows10下安装tensorflow_gpu总结

本文详细记录了在Windows10环境下,配置GTX960m显卡,成功安装tensorflow-gpu1.7-cp36、CUDA9.0、cuDNN7.0.4和anaconda3.5.2的过程。包括检查显卡CUDA兼容性、下载对应版本的安装包、安装anaconda、CUDA和cuDNN,以及解决安装后的版本不兼容问题。
摘要由CSDN通过智能技术生成

windows10下安装tensorflow_gpu总结

我的win10 + gtx960m显卡 + tensorflow-gpu1.7-cp36 + CUDA9.0 + cuDNN7.0.4 + anaconda3.5.2即python3.6 完美安装

一.准备工作:

  1. 1)查看自己显卡支持的驱动CUDA
    查看自己显卡能用的驱动
    2)如何查看显卡支持的CUDA版本
    3)CUDA官方文档
    2) 查看anaconda与Python版本对应
  2. 查看tensorflow版本、python版本、编译器版本、cuDNN、CUDA对应关系
  3. 根据自己的电脑选择安装包下载,一般选择编译好的**.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

https://docs.anaconda.com/anaconda/packages/oldpkglists/ 

release notes

https://docs.anaconda.com/anaconda/reference/release-notes/ 

二.开始安装

  1. 首先安装anaconda
  2. 安装CUDA,注意:
    • 安装时关闭杀毒软件或者一直点击允许
    • 安装中勾选添加环境变量,不然就得自己加
  3. 安装cuDNN,直接将cudnn解压得到的三个文件夹复制到CUDA安装目录下。
  4. 安装tensorflow-gpu,安装所用命令:pip install 文件路径

安装keras

tensorflow和keras版本对应关系 :博客

三.后期问题

  1. 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目录下
  1. ModuleNotFoundError: No module named ‘_pywrap_tensorflow_internal’
    Failed to load the native TensorFlow runtime.
  • 解决步骤:
  • 原因:版本不兼容,降低tensorflow版本
  • 博客
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值