一、使用gpu的三种方式:
1、在python脚本中写入:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
2、在python脚本中写入:
import torch
torch.cuda.set_device(gpu_id)
一、使用gpu的三种方式:
1、在python脚本中写入:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
2、在python脚本中写入:
import torch
torch.cuda.set_device(gpu_id)