Я использовал Miniconda для создания виртуальной среды под названием Spark, версия Torch 2.1.0, версия cuda 12.2.
У меня Pytorch установлен с форума Nvidia, в jetpack6, для aarch64, а именно на этот руль. На Orin я запустил pip install torch-2.1.0-cp310-cp310-linux_aarch64.whl, чтобы установить его.
Мой ~/.bashrc выглядит так:
Код: Выделить всё
# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/home/lyy/miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
eval "$__conda_setup"
else
if [ -f "/home/lyy/miniconda3/etc/profile.d/conda.sh" ]; then
. "/home/lyy/miniconda3/etc/profile.d/conda.sh"
else
export PATH="/home/lyy/miniconda3/bin:$PATH"
fi
fi
unset __conda_setup
export PATH=/usr/local/cuda-12.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH
export C_INCLUDE_PATH=/usr/local/cuda-12.2/include${C_INCLUDE_PATH:+:${C_INCLUDE_PATH}}
export CPLUS_INCLUDE_PATH=/usr/local/cuda-12.2/include${CPLUS_INCLUDE_PATH:+:${CPLUS_INCLUDE_PATH}}
#
Подробнее здесь: [url]https://stackoverflow.com/questions/79187590/how-to-solve-the-undefined-symbol-error-when-importing-torch-scatter-in-python[/url]