威海城市 建设信息网站,网站开发公司会在最后面加上公司,网站建设凭证,wordpress验证码注册在使用 apex 库时#xff0c;按照官方的方式安装后#xff0c;虽然安装成功#xff0c;但调用的时候会报错如下#xff0c;也就是说其实没有成功安装可调用 cuda 的 apex#xff1a;
RuntimeError: apex.optimizers.FusedSGD requires cuda extension我找了很多解决方式按照官方的方式安装后虽然安装成功但调用的时候会报错如下也就是说其实没有成功安装可调用 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 ./