ph4ge/PyTorch-stuff.py

Created April 7, 2025 18:00

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/ph4ge/eadf21a20d414f5b46c72454925c4e24.js"></script>
Save ph4ge/eadf21a20d414f5b46c72454925c4e24 to your computer and use it in GitHub Desktop.

Download ZIP

PyTorch stuff

Raw

PyTorch-stuff.py

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment