reference：linux服务器如何指定gpu以及用量
方法1：在命令行中
CUDA_VISIBLE_DEVICES=0
方法2：在python文件中
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu
linux中指定gpu Id
猜你喜欢
转载自blog.csdn.net/u_7890/article/details/88770383
周排行