安装pytorch出现程序无法定位入口（import torch 报错）的问题解决方案
- 问题描述
- 问题原因
- 解决方法
1 | <b> <font face="微软雅黑" color="red"> QQ:3020889729 小蔡</font></b> |
问题描述
pytorch安装方法如下：(简单选择系统和下载方式即可得到需要的下载命令)
这里把命令放一下，需要的可以直接使用
cpu版的torch下载
1 2 3 4 5 | # cpu版 ——也就是None版 # conda下载方式 conda install pytorch torchvision cpuonly -c pytorch # pip下载方式（较慢——因为后一个下载点是外网）——你可以分开自己单独下载，不过我还是推荐使用conda，方便快捷 pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html |
gpu版下载
conda下载命令
1 2 3 4 5 6 | # cuda==10.2 conda install pytorch torchvision cudatoolkit=10.2 -c pytorch # cuda==10.1 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch # cuda==9.2 conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev |
pip下载命令
1 2 3 4 5 6 | # cuda==10.2 pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html # cuda==10.1 pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html # cuda==9.2 pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html |
使用pytorch安装规范操作安装，但是仍然出现导入包的问题。
即在python和ipython中导入出现问题——这里提供No model is "torch"的解决方法之一。
（关于numpy的版本问题，这里没有考虑，因为按照标准步骤安装的，库应该都是一样的，兼容当前版本的torch的）
问题原因
或许我们是直接进行下载安装的，但是呢？有一个问题可能被忽略了——python版本问题。
版本不兼容，是无法导入的。
解决方法
如果是使用的Anaconda进行包管理，就只需要执行以下命令即可在当前虚拟环境下自动安装库需要的python版本。
1 | codna install python |
如果只是拥有一个python环境（这里指系统下只有一个版本的python且没有Anaconda），那么可能比较麻烦点，需要重新安装对应版本的python。
这里说明一下——关于1.5.0的torch对应的python版本:3.6.6