import torch

device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’)
print(type(device))

import torch
import os
import sys

# 有的地方改的不正去，比如这个ms.set_countext
# https://blog.csdn.net/qq_43215538/article/details/126161578
# 有问题，torch里出来的torch type的cpu
# device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’)

device = ms.set_countext(device_target=’CPU’)
print(type(device))