ubuntu环境下Pytorch安装及No module named 'torch._C'问题解决

安装过程算是比较简单,直接从官网上下载源码,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根目录下运行(本文这个问题)

猜你喜欢

转载自blog.csdn.net/jdk_yxs/article/details/79862993