在你的python命令前加CUDA_VISIBLE_DEVICES=1,指定使用gpu 1
CUDA_VISIBLE_DEVICES=1 python3 -u code2seq_lkf.py

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐