Anaconda3环境PyTorch深度学习框架安装及使用基础

在人工智能与机器学习研究与应用领域，著名的深度学习框架主要有PyTorch、TensorFlow、PaddlePaddle、MXNet，Python/Anaconda语言常与这些深度学习框架联合使用，如Python+PyTorch1.7、Python+TensorFlow2等框架的联合使用。本文以目前常用的Windows10 Anaconda3(Python3.7.0)和Facebook公司的PyTorch1.7深度学习框架为例，介绍Anaconda、PyTorch及Jupyter Notebook IDE Python应用环境的安装及使用测试。
内容提要
一、Anaconda3安装及创建PyTorch虚拟环境
二、PyTorch安装与测试
1.PyTorch安装
2.PyTorch使用测试
3.将PyTorch嵌入到Jupyter Notebook(*: 可选)
三、Jupyter Notebook中PyTorch使用测试

一、Anaconda3安装及创建PyTorch虚拟环境
1. Anaconda3安装
官网一般显示的是最新版本，其下载网址是：https://www.anaconda.com/products/individual，如果想要下载以往版本，可以在下面网址下载：https://repo.anaconda.com/archive/。上面网站下载Anaconda较慢，我们可从清华大学开源软件镜像站https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/下载Anaconda。这里下载Windows版64位Anaconda3(Anaconda3-5.3.0-Windows-x86_64.exe, 631.4 MB, 2018-09-28 06:4，Python对应版本是Python3.7.0）。下载完毕后，运行Anaconda3-5.3.0-Windows-x86_64.exe安装软件，按提示步骤安装Anaconda3。
2. 创建PyTorch虚拟环境
成功安装Anaconda3后，在开始菜单点击【Anaconda Prompt】进入Anaconda命令行窗口。
(1) 选择一个Python版本安装（这里选择python3.7.0版本），并创建PyTorch虚拟环境pytorch：
conda create --name pytorch python=3.7.0 //虚拟环境的名称是pytorch
(2)查看PyTorch虚拟环境是否安装成功:
conda info --envs
我们可以看到有base和pytorch两个环境。
(3) 激活PyTorch虚拟环境pytorch：
activate pytorch
用activate命令激活pytorch虚拟环境。
二、PyTorch安装与测试
在pytorch虚拟环境，顺序输入以下四条命令：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
（说明：在pytorch环境中添加清华镜像，是为了加快下载速度。）
1. PyTorch安装
根据自己需要安装的版本，在pytorch网站https://pytorch.org/寻找安装命令。这里使用如下命令安装CPU版本的PyTorch：
conda install pytorch torchvision cpuonly -c pytorch
（说明：使用清华镜像安装PyTorch需要去掉上面命令后面的“-c pytorch”选项）
2. PyTorch使用测试
（1）激活pytorch：activate pytorch
（2）进入python：python
（3）测试numpy：import numpy
（4）测试pytorch：import torch
（5）显示pytorch版本：

torch.__version__

如果以上步骤都没问题，那么PyTorch安装成功。
3. 将PyTorch嵌入到Jupyter Notebook(*: 可选)
在Anaconda中，我们常使用Jupyter Notebook。将 PyTorch嵌入到 Jupyter Notebook中的命令如下：
(1)激活PyTorch虚拟环境pytorch：activate pytorch
(2)在虚拟环境pytorch中安装Jupyter包：conda install nb_conda
三、Jupyter Notebook中PyTorch使用测试
启动电脑，通过【开始】菜单栏，点击【Jupyter Notebook(pytorch)】进行Jupyter Notebook IDE Python应用环境（见图1）。

图1 Jupyter Notebook中PyTorch使用测试

import torch
print(‘PyTorch =’, torch.__version__)

在图1中，首先用import torch导入PyTorch，然后使用print语句显示PyTorch版本号为目前发行的最新1.7版本， PyTorch在Jupyter Notebook环境中的使用测试成功。接着，我们就可以在Jupyter Notebook IDE Python环境下使用PyTorch框架进行机器学习、科学计算等应用编程了。
对于PyTorch GPU版本的安装要稍微复杂一些，大概需要如下几个步骤：1、安装Nvidia Cuda，2、安装CuDNN，3、安装GPU版本PyTorch，4、测试安装是否成功。
(作者Email联系：yuanzywhu@163.com)
发布日期：2020年11月02日