aurora1625/pytorch.py

Created March 2, 2020 07:55

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/aurora1625/ff190da541ddba162abba6cf8147ba4a.js"></script>
Save aurora1625/ff190da541ddba162abba6cf8147ba4a to your computer and use it in GitHub Desktop.

Download ZIP

Pytorch

Raw

	# PyTorch device
	device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
	X = torch.randn(n_points, 2).to(device)

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