win10环境,pip安装pytorch和torchvision坑总结

1.查看自己的显卡支持cuda版本

win10环境,pip安装pytorch和torchvision坑总结_第1张图片

2.安装pytorch和torchvision

直接前往https://pytorch.org/选择对应的版本操作

win10环境,pip安装pytorch和torchvision坑总结_第2张图片

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 有关的软件。

win10环境,pip安装pytorch和torchvision坑总结_第3张图片

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

你可能感兴趣的:(python)