安装过程算是比较简单,直接从官网上下载源码,https://github.com/pytorch,然后选择从源码来进行安装。(说是可以直接从WHL文件来安装,只不过没有找到。
安装过程的步骤:
git clone https://github.com/pytorch/pytorch
pip install -r requirements.txt(实质上这一步只需要安装pyyaml)。
sudo python setup.py install
安装成功:
import后出现问题
问题解决:
1、发现_C, _dl两个so文件的文件名都太长了,需要进行修改。(此处转载)
采用cp 的方式复制两个文件:
omnisky@omnisky:/usr/local/lib/python3.6/site-packages/torch$ sudo cp _C.cpython-36m-x86_64-linux-gnu.so _C.so
omnisky@omnisky:/usr/local/lib/python3.6/site-packages/torch$ sudo cp _dl.cpython-36m-x86_64-linux-gnu.so _dl.so
发现_C, _dl两个so文件的文件名都太长了,需要进行修改。
采用cp 的方式复制两个文件:
omnisky@omnisky:/usr/local/lib/python3.6/site-packages/torch$ sudo cp _C.cpython-36m-x86_64-linux-gnu.so _C.so
omnisky@omnisky:/usr/local/lib/python3.6/site-packages/torch$ sudo cp _dl.cpython-36m-x86_64-linux-gnu.so _dl.so
2、不要在pytorch根目录下运行(本文这个问题)