aliencaocao/PyTorch_gpu_test.py

Created July 1, 2022 09:51

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

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

Download ZIP

PyTorch test if using GPU

Raw

PyTorch_gpu_test.py

	import torch
	print(torch.cuda.is_available())
	print(torch.cuda.get_device_name(0))
	t = torch.randn(1,2).to('cuda')
	print(t)

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