Neuron Pip Packages within DLAMI Conda Environments FAQ¶
Table of Contents
If the DLAMI Conda Environments do not include the latest Neuron packages, update the packages as follows:
To upgrade Neuron PyTorch:
source activate aws_neuron_pytorch_p36 pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com pip install --upgrade torch-neuron neuron-cc[tensorflow] torchvision
To upgrade Neuron TensorFlow:
source activate aws_neuron_tensorflow_p36 pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com pip install --upgrade tensorflow-neuron tensorboard-neuron neuron-cc
To upgrade Neuron MXNet:
source activate aws_neuron_mxnet_p36 pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com pip install --upgrade mxnet-neuron neuron-cc
To avoid breaking an existing DLAMI enviroment, backup your DLAMI enviroment by creating an AMI from the existing DLAMI environment. Follow instructions at Create an AMI from an Amazon EC2 Instance to save the DLAMI before updating the Neuron Conda packages or upgrading to the latest DLAMI.
Starting with the DLAMI v26.0, the Deep Learning AMI with Conda Options include Neuron Conda packages.
Starting with Neuron SDK 1.14.0, pip packages (Neuron pip packages) are used to install Neuron SDK framework in DLAMI conda environments. To upgrade Neuron SDK framework DLAMI users should use pip upgrade commands instead of conda update commands. Instructions are in How to update to latest Neuron packages in DLAMI Conda Environments?. For more information, see https://aws.amazon.com/blogs/developer/neuron-conda-packages-eol/.
Only Ubuntu 18 and Amazon Linux2 DLAMI are supported (Amazon Linux and Ubuntu 16 are not supported)
Both the DLAMI and Neuron have a monthly release cadence. When there is a new DLAMI release, it will include the latest Neuron Conda packages at the release time. This means that the latest DLAMI version include either the latest Neuron packages or the previous. See Deep Learning AMI (DLAMI) and Neuron versions Matrix for latest DLAMI information.
Update to the latest Neuron packages if the tutorial or the machine learning application you intend to run require a feature or bug fix from the latest Neuron version. See What’s New for information on the latest Neuron version.
You see the version of the running DLAMI by inspecting the README file on the user’s home folder, or at the start of a new terminal session. In the example below the DLAMI version is 35.0
cat ~/README ubuntu@ip-172-31-88-188:~/$ cat ~/README ============================================================================= __| __|_ ) _| ( / Deep Learning AMI (Ubuntu 18.04) Version 35.0 ___|\___|___| =============================================================================