linux安装apex 报错

最新推荐文章于 2023-10-09 16:13:23 发布

原创最新推荐文章于 2023-10-09 16:13:23 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

pytorch 专栏收录该内容

2 篇文章

订阅专栏

在尝试使用pip安装NVIDIA的apex库时遇到错误，具体表现为在执行`pip install -v --no-cache-dir --global-option=--cpp_ext --global-option=--cuda_ext .`命令时失败。解决方法是通过执行`python3 setup.py install`来代替原有的pip安装步骤，成功解决了安装问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用了官方做法

git clone https://github.com/NVIDIA/apex （这一步ok）

cd apex（ok）

最后一步报错：pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

ERROR: Command errored out with exit status 1: /opt/c50017935/condaenv/ViT/bin/python -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-j19rw84j/setup.py'"'"'; __file__='"'"'/tmp/pip-req-build-j19rw84j/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' --cpp_ext --cuda_ext install --record /tmp/pip-record-wd3fgg7r/install-record.txt --single-version-externally-managed --compile --install-headers /opt/c50017935/condaenv/ViT/include/python3.6m/apex Check the logs for full command output.

解决办法

git clone https://github.com/NVIDIA/apex

cd apex

python3 setup.py install