reference:linux服务器如何指定gpu以及用量
方法1:在命令行中
CUDA_VISIBLE_DEVICES=0
方法2:在python文件中
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu
reference:linux服务器如何指定gpu以及用量
方法1:在命令行中
CUDA_VISIBLE_DEVICES=0
方法2:在python文件中
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu