Как установить Нвидиа Апекс ⇐ Python

[Anonymous]

Я пытаюсь установить apex на Colab от Nvidia, но несколько раз не удалось. Я попробовал множество различных решений, включая те, которые предоставлены официальным репозиторием Github. Я также пробовал ответы, представленные здесь.
Каждый раз, когда я пытаюсь, я сталкиваюсь с такой ошибкой `

Код: Выделить всё

torch.__version__  = 1.9.0+cu102

/tmp/pip-req-build-xogkfxc5/setup.py:67: UserWarning: Option --pyprof not specified. Not installing PyProf dependencies!
warnings.warn("Option --pyprof not specified. Not installing PyProf dependencies!")

Compiling cuda extensions with
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_14_21:12:58_PST_2021
Cuda compilation tools, release 11.2, V11.2.152
Build cuda_11.2.r11.2/compiler.29618528_0
from /usr/local/cuda/bin

Traceback (most recent call last):
File "", line 1, in 
File "/tmp/pip-req-build-xogkfxc5/setup.py", line 171, in 
check_cuda_torch_binary_vs_bare_metal(torch.utils.cpp_extension.CUDA_HOME)
File "/tmp/pip-req-build-xogkfxc5/setup.py", line 102, in check_cuda_torch_binary_vs_bare_metal
raise RuntimeError("Cuda extensions are being compiled with a version of Cuda that does " +
RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 10.2.
In some cases, a minor-version mismatch will not cause later errors:  https://github.com/NVIDIA/apex/pull/323#discussion_r287021798.  You can try commenting out this check (at your own risk).
Running setup.py install for apex ... error ERROR: Command errored out with exit status 1: /home/liuyuan/anaconda3/envs/yolox/bin/python3.8 -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-xogkfxc5/setup.py'"'"'; __file__='"'"'/tmp/pip-req-build-xogkfxc5/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' --cpp_ext --cuda_ext install --record /tmp/pip-record-wm_6rdtx/install-record.txt --single-version-externally-managed --compile --install-headers /home/liuyuan/anaconda3/envs/yolox/include/python3.8/apex Check the logs for full command output.

Как видите, я использую последние версии pytorch и cuda, а также рассматривал предыдущие версии, но все решения не увенчались успехом.

Подробнее здесь: https://stackoverflow.com/questions/685 ... vidia-apex