linux安装nvidia/apex---no module named ‘apex‘ 的解决方案
1 官方github (https://github.com/NVIDIA/apex)git clone https://github.com/NVIDIA/apexcd apexpip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./2 来自知乎(https://zhuanl
·
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 ./
更多推荐
已为社区贡献1条内容
所有评论(0)