版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/xwy990/article/details/80995893
环境介绍
Windows 10
cuda 9.1
Python 3.6(virtualenv)
PyTorch 0.4.0
安装过程
从Nvidia官网下载cuda9.1安装包,并全部默认安装选项安装。
从Python官网下载Python3.6安装包并安装。
依照PyTorch官网的说明,使用pip安装PyTorch。
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
pip install torchvision
问题描述
>>> import torch
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Users\username\Code\Python\Test\venv_pytorch\lib\site-packages\torch\__init__.py", line 78, in <module>
from torch._C import *
ImportError: DLL load failed: 找不到指定的模块
解决办法
在安装torchvision时会依赖安装numpy包,日前Windows下用pip安装版本为numpy (1.14.5),主动卸载该版本的numpy
pip uninstall numpy
然后从https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy下载对应版本的numpy+mkl包,并定位到文件路径本地安装,我的下载地址在用户目录的Downloads目录下,故安装命令为
pip install C:\Users\username\Downloads\numpy‑1.14.5+mkl‑cp36‑cp36m‑win_amd64.whl
至此,问题解决,但引发该问题的原因及以上解决办法的原理尚未清楚理解。
另外,PyTorch建议用Anaconda作为Python及其他软件的包管理器,而使用Anaconda未遇到以上问题。