reference:linux服务器如何指定gpu以及用量

方法1:在命令行中

CUDA_VISIBLE_DEVICES=0

 

方法2:在python文件中

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"   #指定第一块gpu

Logo

更多推荐