1.首先安装torch、torchvision、progress、matplotlib
pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install progress -i https://pypi.tuna.tsinghua.edu.cn/simple
2.将cifar.py下的240行的asyncTrue删除
3.出现下面爆红,torchvision版本不匹配,升级torchvision版本
pip install --upgrade torchvision
4.训练TRAINING.MD
5.如果没有安装cuda,就会报下面这个
点击链接,将cuda改为cpu
5.将view改为reshape,继续训练TRAINING.md
6.出现以下报错,将.data[0],[0]改为.item()
7.最后再运行TRAINING.md
选用自己想用的模型