Instalación de la biblioteca Apex, resolución del error de amplificador no instalado (ModuleNotFoundError: ningún módulo llamado 'apex')

Descripción del problema: ¡
Insertar descripción de la imagen aquí
Se trata de una falta de biblioteca apex y es necesario instalarla! Nota: No utilice pip install apex, aunque se puede instalar correctamente, cuando finalmente ejecute el programa, encontrará que todavía hay un error y no se puede utilizar.

Los pasos de instalación específicos son los siguientes:

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

Si el resultado que se muestra en la siguiente figura aparece después de la ejecución, ¡la instalación se realizó correctamente! !
Insertar descripción de la imagen aquí

  • Reponer
# 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'

Es porque el cuda instalado por conda está incompleto y no hay nvcc, ¡la solución es que conda install -c nvidia cuda-nvccla instalación sea exitosa! !

¡Recuerda guardarlo en caso de que no puedas encontrarlo la próxima vez!
Viendo lo serio que hablo, por favor dame un me gusta y vete.

Supongo que te gusta

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