PyTorch 1.0安装失败及GLIBCXX_USE_CXX11_ABI参数调整求助
解决PyTorch 1.0安装失败及GLIBCXX_ABI参数调整问题
一、PyTorch 1.0.0 Conda安装失败的解决方案
- 调整conda频道与源
原命令找不到包可能是频道优先级或源的问题,可尝试添加conda-forge频道后重新安装:
也可切换到国内镜像源(如清华镜像)加速查找,配置后再执行安装命令。conda config --add channels conda-forge conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 - 改用Pip安装
如果conda源无对应包,尝试用pip安装适配Python3.6的版本,需CUDA10.0支持的话可指定CUDA后缀:# 通用版本 pip install torch==1.0.0 torchvision==0.2.1 # CUDA10.0版本 pip install torch==1.0.0+cu100 torchvision==0.2.1 -f https://download.pytorch.org/whl/torch_stable.html - 创建纯净conda环境
现有环境可能存在包冲突,新建Python3.6专属环境再尝试安装:conda create -n torch10_py36 python=3.6 conda activate torch10_py36 conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
二、_GLIBCXX_USE_CXX11_ABI=0的PyTorch 1.0获取及参数调整
- 预编译包获取
官方及公开渠道的PyTorch1.0预编译包基本默认以_GLIBCXX_USE_CXX11_ABI=1编译,现成的=0版本预编译wheel很少见,建议自行源码编译。 - 源码编译PyTorch 1.0(指定ABI参数)
- 克隆并切换到PyTorch1.0.0分支:
git clone https://github.com/pytorch/pytorch.git cd pytorch git checkout v1.0.0 git submodule update --init --recursive - 设置环境变量指定ABI参数后编译安装:
export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=0" export CFLAGS="-D_GLIBCXX_USE_CXX11_ABI=0" python setup.py install
- 克隆并切换到PyTorch1.0.0分支:
- PyTorch1.0.1的参数调整
_GLIBCXX_USE_CXX11_ABI是编译时确定的参数,无法在运行时修改。若必须使用=0,需按照上述源码编译流程,针对PyTorch1.0.1分支重新编译,编译前设置相同的环境变量即可。
内容的提问来源于stack exchange,提问作者Yuan
相关产品推荐
相关产品推荐

