【问题解决】RuntimeError: apex.optimizers.FusedSGD requires cuda extension 问题解决

在使用 apex 库时,按照官方的方式安装后,虽然安装成功,但调用的时候会报错如下,也就是说其实没有成功安装可调用 cuda 的 apex:

RuntimeError: apex.optimizers.FusedSGD requires cuda extension

我找了很多解决方式,最终通过如下方式解决:

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

猜你喜欢

转载自blog.csdn.net/jiaoyangwm/article/details/134582330
今日推荐