示例:运行容器
登录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