verify.py torch test
parent
df79e7e205
commit
876d78d134
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue