查看本机显卡信息
#查看显卡信息,红色框为卡编号
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
所有评论(0)