在 python 命令前加上 CUDA_VISIBLE_DEVICES
可以指定 GPU
CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0
版权声明:本文为weixin_43742643原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
在 python 命令前加上 CUDA_VISIBLE_DEVICES
可以指定 GPU
CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0