目 录CONTENT

文章目录

执行torch_compile模型报错“fatal error: stdatomic.h: No such file or directory”

chenming
2024-12-11 / 0 评论 / 0 点赞 / 14 阅读 / 0 字 / 正在检测是否收录...

现象

如图所示,执行torch.load(model_path)加载这个已经torch.compile(model, mode='max-autotune')好的模型文件,进行推理时,报错。
image-1733886234014

原因

gcc版本太低,本机CENOS7版本如下,需要升级
image-1733886250769

解法

升级到5.2.1版本

wget https://copr.fedoraproject.org/coprs/hhorak/devtoolset-4-rebuild-bootstrap/repo/epel-7/hhorak-devtoolset-4-rebuild-bootstrap-epel-7.repo -O /etc/yum.repos.d/devtools-4.repo
 
 
yum install devtoolset-4-gcc devtoolset-4-binutils devtoolset-4-gcc-c++
 
scl enable devtoolset-4 bash
 
 
ln -s /opt/rh/devtoolset-2/root/usr/bin/* /usr/local/bin/
 
hash -r
 
gcc --version

升级完之后,还没完,这里gcc更新后的版本只在当前控制台临时生效,新开一个控制台还是gcc-4.8.5版本,需要进行第二步操作

替换旧的 gcc

mv /usr/bin/gcc /usr/bin/gcc-4.8.5
ln -s /opt/rh/devtoolset-4/root/bin/gcc /usr/bin/gcc
mv /usr/bin/g++ /usr/bin/g++-4.8.5
ln -s /opt/rh/devtoolset-4/root/bin/g++ /usr/bin/g++
0

评论区