Installation der Apex-Bibliothek, Behebung des Fehlers „Amp nicht installiert“ (ModuleNotFoundError: Kein Modul mit dem Namen „apex“)

Problembeschreibung:
Fügen Sie hier eine Bildbeschreibung ein
Dies ist ein Mangel an Apex-Bibliothek und muss installiert werden! Hinweis: Verwenden Sie nicht pip install apex. Obwohl es erfolgreich installiert werden kann, werden Sie beim endgültigen Ausführen des Programms feststellen, dass immer noch ein Fehler vorliegt und es nicht verwendet werden kann.

Die spezifischen Installationsschritte sind wie folgt:

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

Erscheint nach der Ausführung das in der Abbildung unten dargestellte Ergebnis, ist die Installation erfolgreich! !
Fügen Sie hier eine Bildbeschreibung ein

  • Auffüllen
# 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'

Dies liegt daran, dass der von Conda installierte Cuda unvollständig ist und kein NVCC vorhanden ist. Die Lösung ist, dass conda install -c nvidia cuda-nvccdie Installation erfolgreich ist! !

Denken Sie daran, es zu speichern, falls Sie es das nächste Mal nicht finden können!
Da ich es so ernst meine, gib mir doch bitte ein „Gefällt mir“ und verlasse es.

おすすめ

転載: blog.csdn.net/weixin_43994864/article/details/121411717#comments_28333214