Произошла ошибка при компиляции Pytorch из исходного кода. ⇐ Linux

[Anonymous]

Я пытаюсь скомпилировать Pytorch из исходного кода и столкнулся с этой проблемой, и не могу понять, как с ней справиться.

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

In file included from /usr/local/cuda/bin/../targets/x86_64-linux/include/crt/math_functions.h:10554,
from /usr/local/cuda/bin/../targets/x86_64-linux/include/crt/common_functions.h:303,
from /usr/local/cuda/bin/../targets/x86_64-linux/include/cuda_runtime.h:117,
from :
/usr/include/c++/9/cmath:45:15: fatal error: math.h: No such file or directory
45 | #include_next 
|               ^~~~~~~~
compilation terminated.

Моя машина оснащена двумя графическими процессорами Nvidia 3090.
Система:
Ubuntu: 20.04.2
anaconda3
Python 3.8cuda-11.8
На машине установлены разные версии cuda, поэтому я добавил их в ~/.bashrc:

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

export CUDA_HOME=/usr/local/cuda-11.8
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-11.8/bin:$PATH

Я хотел бы использовать скомпилированный pytorch для тестирования YOLOv5.
Я поместил папку pytorch прямо в домашний каталог /home/lyy.
Я следовал официальному руководству по компиляции Pytorch, а именно следующим шагам:

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

conda create -n yolo python=3.8
conda activate yolo
conda install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
conda install -c pytorch magma-cuda110
conda install -c conda-forge cudatoolkit=11.6
conda install cuda cuda-nvcc -c nvidia/label/cuda-11.6.0

export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:$LD_LIBRARY_PATH
export LDFLAGS="-L$CONDA_PREFIX/lib $LDFLAGS"

export TRACE_KERNEL=1
export USE_CUDNN=1
export DEBUG=1

export CMAKE_LIBRARY_PATH=$CONDA_PREFIX/lib:$CMAKE_LIBRARY_PATH
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}

cd pytorch
python setup.py install

Колесо, которое я пытаюсь построить, — torch-1.12.0a0+git8c4ef23.
Я не знаю, как справиться с ошибкой и искал решения в Интернете, но ни одно из них не дало эффекта.
Я пытался добавить эти окружения:

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

export C_INCLUDE_PATH=
export CPLUS_INCLUDE_PATH=

Но ошибка остается.
Разочарован.

Подробнее здесь: https://stackoverflow.com/questions/793 ... ource-code