docker-compose.yaml可以不写devices的信息,如果要指定哪张卡可以通过environmet:HIP_VISIBLE_DEVICES=1
对于NVIDIA等GPU,在编写docker-compose.yaml时,需要指定GPU设备如下:
services:
openai-whisper-asr-webservice:
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["1"]
capabilities:
- gpu
在使用海光DCU时,如何指定driver类型和capabilities?

暂无数据