深度学习入门:基于PyTorch和TensorFlow的理论与实现
上QQ阅读APP看书,第一时间看更新

3.2 TensorFlow的安装

下面将介绍在Anaconda环境下安装TensorFlow的方法。

首先,我们为TensorFlow创建一个虚拟环境。该虚拟环境应独立于主环境和我们在第2章中创建的PyTorch虚拟环境。

打开Anaconda Prompt,在命令行窗口中输入:

   > conda create --name tensorflow python=3.5

注意,这里的tensorflow是虚拟环境的名称。虚拟环境创建完成之后,可以输入以下命令,进入虚拟环境tensorflow:

   > activate tensorflow

如果不想使用TensorFlow,可以输入deactivate tensorflow,关闭当前虚拟环境。

进入该虚拟环境之后,我们就可以安装TensorFlow了。如果安装CPU版本的TensorFlow,直接在Anaconda Prompt中输入一行命令即可:

   conda install tensorflow

如果安装GPU版本的TensorFlow,计算机中需要有一块NVIDIA的GPU显卡并安装了显卡驱动。在安装TensorFlow之前,需要提前安装CUDA和CUDNN。提前安装工作准备好之后,直接在Anaconda Prompt中输入一行命令即可:

   conda install tensorflow-gpu

因为安装GPU版本的TensorFlow需要有硬件支持,而且准备工作较多,因此,作为入门学习,本书推荐安装CPU版本的TensorFlow。

完成了TensorFlow的安装之后,接下来,要在tensorflow这个虚拟环境中安装Jupyter和其他所需的Python库。与PyTorch的安装过程相同,直接使用conda命令安装即可,这里不再赘述。

安装工作完成后,首先我们来启动Jupyter Notebook。打开Anaconda Navigator,在Anaconda Navigator界面的Applications on下拉列表框中选择tensorflow,启动该环境下的Jupyter Notebook,如图3-1所示。

图3-1 切换至虚拟环境tensorflow

打开Jupyter Notebook之后,输入import语句,如果没有报错,就说明TensorFlow已经安装成功了。也可以在Jupyter Notebook查看TensorFlow的版本,如图3-2所示。

图3-2 导入TensorFlow库并查看TensorFlow的版本