快速入门指南#

注意

本页面假定您已安装并设置 前提软件 (Docker, NGC CLI, NGC 注册表访问)。

  1. 使用以下命令拉取 NIM 容器。

docker pull nvcr.io/nim/nvidia/molmim:1.0.0

  1. 使用以下命令运行 NIM 容器。

docker run --rm -it --name molmim --runtime=nvidia \
    -e CUDA_VISIBLE_DEVICES=0 \
    -e NGC_CLI_API_KEY \
    -p 8000:8000 \
    nvcr.io/nim/nvidia/molmim:1.0.0

此命令启动 NIM 容器并暴露端口 8000，供用户与 NIM 交互。

  1. 打开一个新的终端，保持当前终端打开并运行服务。

  2. 在新终端中，等待健康检查端点返回 {"status":"ready"} 后再继续。这可能需要几分钟。您可以使用以下命令查询健康检查。

curl -X 'GET' \
    'http://127.0.0.1:8000/v1/health/ready' \
    -H 'accept: application/json'

  1. 运行推理，从分子的 SMILES 字符串表示中获取嵌入，并将输出保存到 output.json，使用以下命令。

curl -X 'POST' \
    'http://127.0.0.1:8000/embedding' \
    -H 'accept: application/json' \
    -H 'Content-Type: application/json' \
    -d '{"sequences": ["CC(Cc1ccc(cc1)C(C(=O)O)C)C"]}' > output.json

  1. 查看输出。您可以使用 cat 工具将输出打印到命令行，如下面的命令所示。

cat output.json

但是，我们建议安装 jq，这是一个命令行工具，可以格式化 JSON 以提高可读性。您可以使用 jq 可视化文件中的输出

jq . output.json

或者您可以将输出直接管道传输到 jq，如下面的命令所示

curl -X 'POST' \
    'http://127.0.0.1:8000/embedding' \
    -H 'accept: application/json' \
    -H 'Content-Type: application/json' \
    -d '{"sequences": ["CC(Cc1ccc(cc1)C(C(=O)O)C)C"]}' > output.json