当前位置:   article > 正文

docker container 指定gpu设备_docker-compose gpus

docker-compose gpus

1, 在yaml中   Turn on GPU access with Docker Compose | Docker Docs

Example of a Compose file for running a service with access to 1 GPU device:

  1. services:
  2. test:
  3. image: nvidia/cuda:12.3.1-base-ubuntu20.04
  4. command: nvidia-smi
  5. deploy:
  6. resources:
  7. reservations:
  8. devices:
  9. - driver: nvidia
  10. count: 1
  11. capabilities: [gpu]

Access specific devices

To allow access only to GPU-0 and GPU-3 devices:

  1. services:
  2. test:
  3. image: tensorflow/tensorflow:latest-gpu
  4. command: python -c "import tensorflow as tf;tf.test.gpu_device_name()"
  5. deploy:
  6. resources:
  7. reservations:
  8. devices:
  9. - driver: nvidia
  10. device_ids: ['0', '3']
  11. capabilities: [gpu]

2,   执行以下命令

docker run --rm -ti --gpus device=1 -p 8080:8080 -e DEBUG=true -e MODELS_PATH=/models -e THREADS=1 -v $PWD/models:/models quay.io/go-skynet/local-ai:master-cublas-cuda12

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/一键难忘520/article/detail/849428
推荐阅读
相关标签
  

闽ICP备14008679号