1
0
Fork 0

Build Torch on ppc64le

main
root 2023-10-17 12:53:40 -06:00
parent 479352eb4e
commit a6caa81185
1 changed files with 42 additions and 0 deletions

View File

@ -157,3 +157,45 @@ sudo apt purge libideep-dev
TORCH_USE_CUDA_DSA
\end{minted}
\section{Torch POWER}
Running pytorch on ppc64le.
In particular, the Raptor Computing Talos II motherboard.
\begin{minted}{sh}
# Set the compilers, gcc-12 is nogo.
# Using ccache...
export CMAKE_C_COMPILER=/usr/lib/ccache/gcc-11
export CMAKE_CXX_COMPILER=/usr/lib/ccache/g++-11
# First time clone:
git clone --recursive https://github.com/pytorch/pytorch
# And subsequently:
cd pytorch/
# Clean behind your ears:
git checkout main
git reset --hard HEAD
git clean -ff
git pull
git submodule update --init
git submodule update
# Get latest release
git tag -l | sort -V | grep -v -e rc | tail -1
git checkout `git tag -l | sort -V | grep -v -e rc | tail -1`
# Virtual environment for Python, such as:
source deactivate
rm -rf venv
pyenv local 3.12
virtualenv -p 3.12 venv
source venv/bin/activate
pip install -U setuptools wheel
pip install -r requirements.txt
# Set options, install wanted dependencies:
ccmake build -DCUDAToolkit_INCLUDE_DIR=/usr/include
# Install into virtual environment
python setup.py install
\end{minted}