问题描述
在运行Pytorch时报错,提示当前使用的c++和torch编译用的不一致。尝试了几个方法去解决。
1. 设置默认的编译器
echo “export CXX=g++” >> ~/.bashrc
source ~/.bashrc
这个是在网上找到的,https://github.com/NVIDIA/apex/issues/974 。但没有解决我的问题。
2. 定位问题所在
(我故意在cpp_extension.py写入错误指令)
发现报错来自于自己调用.cpp文件的一个py文件,于是我在该文件中写入:
import os
os.environ[‘CXX’] = ‘g++’ # Set the CXX environment variable to g++
通过设置CXX环境,这样就解决了我的问题。