1 官方github (https://github.com/NVIDIA/apex

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

报错是因为基础环境有误,参考
https://blog.csdn.net/justforacm/article/details/115428583

2 来自知乎(linux安装nvidia/apex - 知乎

git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install
python setup.py install --cuda_ext --cpp_ext

3. 如果还不行

export TORCH_CUDA_ARCH_LIST="compute capability" 
conda install -c nvidia cuda-nvcc

cd apex-22.04-dev
pip install -r requirements.txt
pip install -r requirements_dev.txt
pip install -v --disable-pip-version-check --no-cache-dir ./

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐