Activating PyTorch Install PyTorch's Nightly Build (experimental)More Tutorials

PyTorch

Activating PyTorch

When a stable Conda package of a framework is released, it's tested and pre-installed on the DLAMI. If you want to run the latest, untested nightly build, you can Install PyTorch's Nightly Build (experimental) manually.

To activate the currently installed framework, follow these instructions on your Deep Learning AMI with Conda.

For PyTorch on Python 3 with CUDA 10 and MKL-DNN, run this command:


$ source activate pytorch_p36

For PyTorch on Python 2 with CUDA 10 and MKL-DNN, run this command:


$ source activate pytorch_p27

Start the iPython terminal.


(pytorch_p36)$ ipython

Run a quick PyTorch program.


import torch
x = torch.rand(5, 3)
print(x)
print(x.size())
y = torch.rand(5, 3)
print(torch.add(x, y))

You should see the initial random array printed, then its size, and then the addition of another random array.

Install PyTorch's Nightly Build (experimental)

How to install PyTorch from a nightly build

You can install the latest PyTorch build into either or both of the PyTorch Conda environments on your Deep Learning AMI with Conda.

- (Option for Python 3) - Activate the Python 3 PyTorch environment:
```
$ source activate pytorch_p36
```
- (Option for Python 2) - Activate the Python 2 PyTorch environment:
```
$ source activate pytorch_p27
```
The remaining steps assume you are using the pytorch_p36 environment. Remove the currently installed PyTorch:
```
(pytorch_p36)$ pip uninstall torch
```

(Option for GPU instances) - Install the latest nightly build of PyTorch with CUDA 10.0:


(pytorch_p36)$ pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html

(Option for CPU instances) - Install the latest nightly build of PyTorch for instances with no GPUs:
```
(pytorch_p36)$ pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
```

To verify you have successfully installed latest nightly build, start the IPython terminal and check the version of PyTorch.
```
(pytorch_p36)$ ipython
```
```
import torch
print (torch.__version__)
```
The output should print something similar to 1.0.0.dev20180922

To verify that the PyTorch nightly build works well with the MNIST example, you can run a test script from PyTorch's examples repository:


(pytorch_p36)$ cd ~
(pytorch_p36)$ git clone https://github.com/pytorch/examples.git pytorch_examples
(pytorch_p36)$ cd pytorch_examples/mnist
(pytorch_p36)$ python main.py || exit 1