最近在学习pytorch转ONNX模型,调通了代码,想验证下生成的ONNX文件是否正确,参照pytorch官网教程,需要先安装onnx库:
可能因为是国外的源,所以一直下载不成功,换了国内的源也没用(好像说-c这个参数指定的是在国外)
后面就打算直接去下载安装包,本地安装。
先去https://anaconda.org/conda-forge搜索onnx库的安装包,如下图:
点击onnx
由于当前页面显示的是下载onnx的命令,而我们需要下载安装包,所以选择红框中的“File”:
注意:根据自己的系统和python版本进行选择,不要选错了哦~
我的是win64,python3.6,所以选择图中所示的安装包,点击进行下载。
下载到本地后,将其拷贝到所需安装目录下。
如果你不需要安装在自建环境中,而是安装在anaconda中,那么将tar.bz2文件复制到:Anaconda/pkgs文件夹下,在此处打开cmd进行安装;
而我的是需要安装在pytorch环境中(pytorch是我自建环境的环境名),那么需要将安装包拷贝到环境目录下,然后进入Anaconda Prompt
激活环境:activate pytorch
cd到安装包所在目录下,执行下述命令即可
conda install --use-local onnx-1.8.1-py36h6d34f3e_0.tar.bz2
安装成功!
打开pycharm,测试onnx库是否安装成功
新建python file,输入import onnx,可以看到安装成功,大功告成!
如果还不行的话,那就找个夜深人静的时候,再试试:
conda install -c conda-forge onnx
说不定就可以啦!!!
希望可以成功哦~~~