问题描述:
在python程序中导入torch时显示没有此模块,并且不能直接使用pip安装torch
我的电脑版本
cu90: cuda9.0
torch-1.1.0 : torch版本 1.1.0
cp37 : python版本3.7
win: windows 系统
按照自己的电脑版本下载即可,方法通用
解决方案:
- 查找对应torch版本
打开
pytorch官网
(https://pytorch.org/),从链接选择指定版本
CUDA版本查询方法如下
- 查询CUDA版本的方法:
打开控制面板——>查看方式选择小图标——>选择NVIDIA控制面板
选择系统信息
选择 “组件”,即可看到型号
- 进入 https://download.pytorch.org/whl/torch_stable.html 按照上面查询的版本下载两个文件(迅雷下载较快)
cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl
cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
如:
cu90即 CUDA9.0
cp37即 python3.7
按自己电脑版本选择即可
4.下载到本地后,使用terminal,安装
分别输入:
pip –default-timeout=1000 install
E:\Python\Scripts\torch-1.1.0-cp37-cp37m-win_amd64.whl
(文件位置)
pip –default-timeout=1000 install
E:\Python\Scripts\torchvision-0.3.0-cp37-cp37m-win_amd64.whl
(文件位置)
安装结束