1.直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python my_script.py
2.python代码中设定:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="2"
3.使用函数 set_device
import torch
torch.cuda.set_device(id)
1.直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python my_script.py
2.python代码中设定:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="2"
3.使用函数 set_device
import torch
torch.cuda.set_device(id)