Windows上深度学习框架Pytorch与Tensorflow的安装

深度学习大热,如何搭建深度学习所需的编程环境呢?本文将提供一个简明教程。
首先我们来了解深度学习编程相关的几个概念:
编程语言/编译器:Python。Python的特点是“用最少的代码干最多的事”。Python 2即将在2020年停止更新,所以现在学习Python 3是最好的选择。
编辑器:Pycharm。非常好用的Python集成开发环境。
深度学习框架1:Pytorch。Facebook支持的深度学习框架,学习曲线较为平缓,非常适合初学者、研究人员。
深度学习框架2:Tensorflow。谷歌支持的深度学习框架,学习曲线较为陡峭,目前更适合工程人员。TF 2.0版本正在开发中,相对于之前的TF 1.x会有较大的调整和变化。
图形处理器:主流深度学习框架依赖Nvidia GPU的CUDA和CuDNN。如果你的电脑有Nvidia GPU,可以加速深度学习程序的执行。
接下来我们将介绍深度学习环境在Windows上的安装方法。
Windows上安装深度学习环境
Windows系统是目前占有率最高的桌面操作系统,本身不包含Python环境。因此我们首先需要下载Python环境,这里推荐的Python环境管理工具是Anaconda。
首先从官网下载安装Anaconda (Python 3.7版本)
Windows上深度学习框架Pytorch与Tensorflow的安装
如果机器有Nvidia GPU,先从Nvidia官网下载安装CUDA 10.0。
Windows上深度学习框架Pytorch与Tensorflow的安装
然后下载CuDNN 7,解压并将 .h文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include,将.lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64``。
安装Pytorch(需要联网):
如果有Nvidia GPU,在命令提示符中执行

  1. conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
    否则安装CPU版,在命令提示符中执行
    1.conda install pytorch-cpu torchvision-cpu -c pytorch
    安装Tensorflow(需要联网):
    如果有Nvidia GPU,在命令提示符中执行
    1.conda install -c anaconda tensorflow-gpu
    否则安装CPU版,在命令提示符中执行
    1.conda install -c anaconda tensorflow
    下载安装Pycharm,不需要Web开发功能的我们选择Community版本即可,是免费的。
    Windows上深度学习框架Pytorch与Tensorflow的安装
    测试环境
    打开命令提示符,执行
    1.python
    2.import torch
    3.import tensorflow as tf
    如果没有错误,则表明深度学习框架Pytorch和Tensorflow已经安装成功。

配置Pycharm找到Python
为了让编辑器Pycharm找到Python编译器,我们需要进行一定的配置。
打开Pycharm,打开菜单 File->Default Settings->Project Intepreter,点击设置图标(图中红色圈出),输入本地python.exe的路径。
一路点击OK保存设置。
Windows上深度学习框架Pytorch与Tensorflow的安装
至此,你已经可以用Pycharm开始开发深度学习程序了。访问Pytorch官网教程,开始学习吧!