在Windows下使用pip安装TensorFlow,含各种问题解决办法,亲测有效

前言

本文使用pip在window下安装tensorflow 2,由于tensorflow 2版本默认支持GPU,所以需要额外做GPU支持的相关准备。

系统要求

  • Python 3.5-3.7(64位)
  • pip 19.0或更高版本
  • Windows 7或更高版本(64位)

所需环境准备

  • 安装适用于 Visual Studio 2015、2017 和 2019 的 Microsoft Visual C++ 可再发行软件包。我使用的是这个2015官网下载,下载后直接安装即可。官网目前不提供2019可再发行软件包,可自行选择2015或2017下载,下载VS实在是太大了。

  • 在Windows上启用长路径
    按下键盘上的Win+R键打开运行命令框;
    输入regedit然后回车打开注册表编辑器;
    转到以下注册表项:
    HKEY_LOCAL_MACHINE\System\CurrentControlSet\Policies
    注册表选项
    在右侧,创建一个名为LongPathsEnabled的新的32位DWORD值。将其值数据设置为1。重启电脑即可。

  • CUDA® 工具包:TensorFlow 支持 CUDA 10.1(TensorFlow 2.1.0 及更高版本)
    注意版本一定要匹配,版本高于或低于都会出现问题。
    下载时最好选择本地安装版本,这样下载和安装速度会快很多。

  • cuDNN SDK(7.6 及更高版本)
    下载好后解压并进行如下复制:
    (1)将cudnn\bin\cudnn64_7.dll 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\,然后将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\cudnn64_7.dll添加到Path环境变量中
    (2)将 cudnn\include\cudnn.h 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
    (3)将cudnn\lib\x64\cudnn.lib 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64\

安装Tensorflow

由于Tensorflow包比较大,这里建议先下载好whl文件,然后直接进行安装

  • 官网下载包文件,
  • 打开命令行窗口,切换目录到whl文件的同级目录下,执行以下命令进行安装
pip install tensorflow-2.1.0-cp37-cp37m-win_amd64.whl
  • 安装成功后,执行如下代码测试
python
import tensorflow as tf

参考文档

  • 官网:https://www.tensorflow.org/install/pip
  • 安装错误解析方案:https://www.tensorflow.org/install/errors
  • https://stackoverflow.com/questions/43942185/failed-to-load-the-native-tensorflow-runtime-python-3-5-2
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值