diff --git a/src/Pytorch.tex b/src/Pytorch.tex index 3f82731..c5bcb15 100644 --- a/src/Pytorch.tex +++ b/src/Pytorch.tex @@ -158,6 +158,24 @@ TORCH_USE_CUDA_DSA \end{minted} +\section{Pytorch Test} +Quick and dirty verify.py script to run after installing Pytorch to make +sure everything is ok: + +\begin{minted}{python} +#!/usr/bin/env python3 + +import torch +x = torch.rand(5, 3) +print(x) + +torch.cuda.set_device(0) +print("CUDA:", torch.cuda.is_available()) + +print("Pytorch Version:", torch.__version__) +\end{minted} + + \section{Torch POWER} Running pytorch on ppc64le. In particular, the Raptor Computing Talos II motherboard.