环境
- ubuntu
- docker + nvidia-docker
步骤
注册账号
官网地址https://ngc.nvidia.com/,注册后登录
生成API Key(重要)
在左侧选择CONFIGURATION
,然后在Generate API Key
中点击Get API Key
,点击右上角绿色按钮Generate API Key
准备拉取镜像
在左侧选择’ACCELERATED SOFTWARE’,然后在众多软件中选择自己需要的那一款,这里我们选择NVCaffe,在Tags选择我们需要的版本,然后复制拉取命令,例如docker pull nvcr.io/nvidia/caffe:19.04-py2
在docker中登录
注意我们需要在docker中登录nvcr.io
,否则拉取镜像时会遇到"unauthorized: authentication required"
问题。
登录命令如下:
$ docker login nvcr.io
Username: $oauthtoken
Password: <Your Key>
如果觉得Key太长