如何使用类似 CUDA_VISIBLE_DEVICES 的效果?
收藏回复举报
如何使用类似 CUDA_VISIBLE_DEVICES 的效果?
t('forum.solved') 已解决
发表于2024-08-02 16:39:56
0 查看
我的 npu-smi info 显示结果中 npu 这一列显示的 5 6 7 8 (我有4张卡)。我尝试了使用 ASCEND_VISIBLE_DEVICES=0 / 1 / 5 / 6 test.sh 四种命令来运行程序,默认跑的都是第一张卡(test.sh 是我要跑的程序),这个是哪里有问题吗?

我要发帖子