Instalação da biblioteca Apex, resolvendo erro de amplificador não instalado (ModuleNotFoundError: Nenhum módulo chamado 'apex')

Descrição do problema:
Insira a descrição da imagem aqui
Falta a biblioteca apex e precisa ser instalada! Nota: Não use pip install apex, embora possa ser instalado com sucesso, quando você finalmente executar o programa, descobrirá que ainda há um erro e não pode ser usado.

As etapas específicas de instalação são as seguintes:

# 执行git下载apex文件夹时,如果下载太慢,可以手动进入网址https://github.com/NVIDIA/apex下下载解压后执行后续操作
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install

Se o resultado mostrado na figura abaixo aparecer após a execução, a instalação foi bem-sucedida! !
Insira a descrição da imagem aqui

  • Reabastecimento
# error
Traceback (most recent call last):
File "setup.py", line 35, in <module>
_, bare_metal_major, _ = get_cuda_bare_metal_version(CUDA_HOME)
File "setup.py", line 14, in get_cuda_bare_metal_version
raw_output = subprocess.check_output([cuda_dir + "/bin/nvcc", "-V"], universal_newlines=True)
TypeError: unsupported operand type(s) for +: 'NoneType' and 'str'

É porque o cuda instalado pelo conda está incompleto e não existe nvcc, a solução é que conda install -c nvidia cuda-nvcca instalação seja bem sucedida! !

Lembre-se de salvá-lo caso não consiga encontrá-lo na próxima vez!
Vendo o quão sério estou falando, por favor me dê um like e vá embora.

Acho que você gosta

Origin blog.csdn.net/weixin_43994864/article/details/121411717#comments_28333214
Recomendado
Clasificación