[MNIST06]pytorch

Posted 2021-09-28 AIplusX

写在前面

之前模型的识别率一直上不去,所以我一直在看我的代码和公式,真的找不出问题来,于是我发现了新手从头开始写模型的坏处,那就是调试起来实在是太麻烦了,远没有带框架的程序调试起来方便.

在多种模型框架之间徘徊了一段时间之后,我最终选择了pytorch,因为他能兼容numpy,使用起来也较为简单,那么这篇文章我就主要分享pytorch环境的安装以及联合GPU的调试.

主要内容

主要内容在我的古月居博客：
[MNIST06]pytorch

pytorch环境安装

我是用conda安装的pytorch,命令如下:

conda install torch
conda install pytorch
conda install torchvision

但是我在安装的时候终端报了这个错误:

pytorch环境测试

我们先来测试一下pytorch环境,我用的是jupyter notebook,首先导入必要的工具包:

from __future__ import print_function
import torch
import numpy as np
import  time

主要内容

主要内容在我的古月居博客：
[MNIST06]pytorch