[ec2-user@ip-172-31-40-200 Notebooks]$ conda install pytorch -c soumith
Fetching package metadata .....
......
Solving package specifications: .

Package plan for installation in environment /home/ec2-user/anaconda3:

The following NEW packages will be INSTALLED:

    pytorch: 0.1.11-py360_4cu80 soumith [cuda80]

Proceed ([y]/n)?
y

ec2-user@ip-172-31-40-200 Notebooks]$ conda list | grep pytorch
    pytorch                   0.1.11              py360_4cu80  [cuda80]  soumith