查看本机显卡信息

#查看显卡信息,红色框为卡编号
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
Logo

新一代开源开发者平台 GitCode,通过集成代码托管服务、代码仓库以及可信赖的开源组件库,让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐