Build Torch on ppc64le
parent
479352eb4e
commit
a6caa81185
|
@ -157,3 +157,45 @@ sudo apt purge libideep-dev
|
||||||
TORCH_USE_CUDA_DSA
|
TORCH_USE_CUDA_DSA
|
||||||
\end{minted}
|
\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}
|
||||||
|
|
Loading…
Reference in New Issue