前言
本文使用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