pytorch深度学习框架--gpu和cpu的选择

pytorch深度学习框架–gpu和cpu的选择

基于pytorch框架，最近实现了一个简单的手写数字识别的程序，我安装的pytorch是gpu版(你也可以安装cpu版本的，根据个人需要)，这里我介绍pytorch的gpu版本和cpu版本的安装以及训练手写数字识别时gpu和cpu之间的切换。
1、pytorch的安装
1.1 pytorch(带有gpu)安装
首先进入pytorch官网，选择自己所需要的版本，这里我选择的版本如下图所示。
然后打开anaconda Prompt，首先输入：conda activate py3激活py3（解释一下为什么是py3，因为我之前装的是python3.6,创建的名字为py3），然后输入：conda install pytorch torchvision cudatoolkit=9.0 -c pytorch安装pytorch，等待安装就好，如下图所示。


1.2 pytorch(无gpu)安装
这时CUDA选择none即可

打开anaconda终端，首先激活py3,然后输入这个命令：conda install pytorch-cpu torchvision-cpu -c pytorch，等待安装就好，如下图所示，

1.3 测试是否安装成功
首先cmd打开终端，输入python即可查看当前安装的python的版本，然后import torch 等待几