在你的python命令前加CUDA_VISIBLE_DEVICES=1,指定使用gpu 1
CUDA_VISIBLE_DEVICES=1 python3 -u code2seq_lkf.py
简单的指定程序使用某个gpu
最新推荐文章于 2024-07-26 17:23:36 发布
在你的python命令前加CUDA_VISIBLE_DEVICES=1,指定使用gpu 1
CUDA_VISIBLE_DEVICES=1 python3 -u code2seq_lkf.py