RuntimeError: apex.optimizers.FusedSGD requires cuda extension

REALLYAI

于 2022-12-05 15:24:40 发布

阅读量1.8k

点赞数

分类专栏： python 文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/qq_42037273/article/details/128187470

版权

python 专栏收录该内容

42 篇文章

订阅专栏

参考https://zhuanlan.zhihu.com/p/80386137
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option=“–pyprof” --global-option=“–cpp_ext” --global-option=“–cuda_ext” ./
就可以了
我最初用的python setup.py install 替换的上述第三步，报错了，导致apex没有cuda版本…

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--pyprof" --global-option="--cpp_ext" --global-option="--cuda_ext" ./