《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

 

posted on 2020-06-10 18:36  适当大  阅读(198)  评论(0编辑  收藏  举报

导航