我想在Bianbu上使用pytorch 2.6, SpacemiT的pypi源上并没有这个版本的。另外，我使用pip install torch -i https://git.spacemit.com/api/v4/projects/33/packages/pypi/simple 安装的pytorch似乎有点问题，执行一些程序时会提示from torch._C._distributed_c10d import (
ModuleNotFoundError: No module named ‘torch._C._distributed_c10d’; ‘torch._C’ is not a package。
于是，我决定自己从源码编译安装pytorch。
首先下载源码
wget https://github.com/pytorch/pytorch/releases/download/v2.6.0/pytorch-v2.6.0.tar.gz
然后解压
tar xvzf pytorch-v2.6.0.tar.gz
接着安装依赖
cd pytorch-v2.6.0/
pip install ninja
pip install -r requirements.txt
pip uninstall cmake # 使用系统的cmake
最后进行编译安装
python -m pip install --no-build-isolation -v -e .
耐心等待它编译完就可以使用了。注意，由于是采用可编辑的安装模式，安装后千万不要删除这个目录的代码文件。
1 个赞