lijunzh/pytorch.yml

Last active March 30, 2020 20:33

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

Download ZIP

Conda environment file to set up latest PyTorch on Ubuntu

Raw

pytorch.yml

	name: pytorch
	channels:
	- pytorch
	dependencies:
	- pip
	- pytorch
	- torchvision
	- cudatoolkit=10.1

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