《Tensorflow实战Google深度学习框架第二版》在win10中的环境搭建
1 安装Anaconda,进入Anaconda Prompt
2 创建tensorflow1.4.0的环境,指定Python版本为3.5。然后激活该环境。
conda create -n TF1.4.0 python=3.5
conada activate TF1.4.0
3 安装Nvidia的SDK 8.0版本(cudatoolkit 8.0)和相应的Nvidia深度学习软件包6.0版本(cudnn6.0.21)
conda install cudatoolkit=8.0 conda install cudnn=6.0.21
注意,可以在https://anaconda.org上的搜索框里面搜索相应软件包的具体版本,里面有具体的安装命令。这里只安装cudnn6.0.21的版本。
4 安装Tensorflow1.4.0
pip install tensorflow==1.4.0
安装完成后,运行一下代码,若结果返回Tensorflow的版本号'1.4.0'则环境搭建成功。
python import tensorflow as tf tf.__version__
最后可能会显示一些关于numpy的警告信息,这是由于numpy版本较高的缘故,可以卸载环境中默认安装的numpy包,安装版本为1.16的numpy包。运行代码如下:
pip uninstall numpy pip install numpy==1.16