1
0
Fork 0

verify.py torch test

main
root 2023-10-17 13:14:23 -06:00
parent df79e7e205
commit 876d78d134
1 changed files with 18 additions and 0 deletions

View File

@ -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.