docker指定使用某几张显卡/某几个GPU

查看本机显卡信息

#查看显卡信息，红色框为卡编号
nvidia-smi

启动容器指定显卡

#使用第3,4张卡
--gpus='"device=3,4"'

示例：

docker run -ti --gpus='"device=3,4"' \
-v /data/common_data/ASR:/workspace/data \
--tmpfs /tmp \
--entrypoint=/bin/bash \
--net=host \
--ipc=host \
seannaren/deepspeech.pytorch:latest