1.查看自己的显卡支持cuda版本
2.安装pytorch和torchvision
直接前往https://pytorch.org/选择对应的版本操作
3.运行代码出现报错:
from torch._C import *
ImportError: DLL load failed: 找不到指定的模块。
解决:
打开https://anaconda.org/anaconda/intel-openmp/files,下载 win-64/intel-openmp-2018.0.0-8.tar.bz2 并解压 (根据自己的系统下载,我的系统的 win64 位)。
打开解压后的文件夹intel-openmp-2018.0.0-8 中的 Library/bin,将其中的两个 .dll 文件复制并放在一个新的文件夹中。
将 .dll 文件的文件夹路径添加到系统的环境变量中,重启 Python 有关的软件。
4.搭建图像分类网络参考代码
https://cloud.tencent.com/developer/article/1383124
5.解决CIFAR10下载过慢的问题
第一步,手动下载到本地
第二步,修改Python36\Lib\site-packages\torchvision\datasets下的cifar.py文件。将第35行的链接https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz修改为本地地址,例如:
file:///F:/codes/neural_net/cifar-10-python.tar
附注:强制安装最新版本pip
python2 -m pip install --upgrade --force-reinstall pip