在 /home/$USER/milvus/conf 路径下的 server_config.yaml 文件中,你可以为建立索引以及检索分配 GPU。如下代码所示:

    1. gpu:
    2. enable: true
    3. cache_size: 10GB
    4. gpu_search_threshold: 0
    5. search_devices:
    6. - gpu0
    7. - gpu1
    8. build_index_devices:
    9. - gpu2
    10. - gpu3
    gpu 0, 1, 2, 3 是分配给 Docker 容器的 GPU 中的前四个。