Build Torch on ppc64le
parent
479352eb4e
commit
a6caa81185
|
@ -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}
|
||||
|
|
Loading…
Reference in New Issue