如果只是要指定第三张卡跑任务，可以在启动时指定环境变量HIP_VISIBLE_DEVICES=2；如果希望容器内只有一个卡可见，可以把--device=/dev/dri这个选项改成--device=/dev/dri/render{按照编号推测卡号}，并不要遗漏--device=/dev/mkfd
0
2
0
0
标题
2025-03-21 16:59:05
177
docker启动pytorch基础环境，怎么指定DCU
docker run -it --shm-size=64G --network host -v /root/demo/model/bge/bge-m3:/home/bge-m3 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name bge-m3 <dee41741fb40> bash
比如在第3块k100-ai上跑这个pytorch基础环境
全部评论（2）
只看楼主
R
2025/3/23
T
进入容器，设置环境变量HIP_VISIBLE_DEVICES=2，默认第一张卡是0
2025/3/22
没有更多内容了
--
楼主
0 点赞
0 帖子
楼主热帖
暂无数据常见问题新帖
暂无数据