示例:运行容器

登录NGC容器注册表

如果通过SSH登录到虚拟机示例时使用了API密钥,请忽略此部分。否则,需要执行如下操作。

用下面的Docker命令登录到NGC容器注册表:

docker login nvcr.io

系统会提示您输入用户名和密码。按照所示输入”$oauthtoken”,并输入您在设置NGC账户时获取的NGC API密钥:

Username: $oauthtoken
Password: <Your NGC API Key>

完成后,您可以运行Docker命令并从VM实例访问NGC容器注册表。

示例:用PyTorch容器运行MNIST训练

一旦登录到阿里云GN5实例,您就可以在PyTorch下运行MNIST例子。

注意:PyTorch示例将从网上下载MNIST数据集。

获取并运行PyTorch容器:

docker pull nvcr.io/nvidia/pytorch:18.02.py3
nvidia-docker run --rm -it nvcr.io/nvidia/pytorch:18.02.py3

运行MNIST示例:

cd /opt/pytorch/examples/mnist
python main.py

示例:用TensorFlow容器运行MNIST训练

一旦登录到阿里云 GN5实例,您就可以在TensorFlow下运行MNIST例子。

注意:TensorFlow自带的示例将从网上下载MNIST数据集。

获取并运行TensorFlow容器:

docker pull nvcr.io/nvidia/tensorflow:18.02.py3 
nvidia-docker run --rm -it nvcr.io/nvidia/tensorflow:18.02.py3

可参考: https://www.tensorflow.org/get_started/mnist/beginners

运行MNIST_with_summaries示例:

cd /opt/tensorflow/tensorflow/examples/tutorials/mnist 
python mnist_with_summaries.py