查看 GPU 是否可用,返回 True 则具有能够使用的 GPU
torch.cuda.is_available()
查看本机 GPU 数量
import torch
n_gpu = torch.cuda.device_count()
print(n_gpu)
将模型部署到 GPU
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
精于心,简于形
查看 GPU 是否可用,返回 True 则具有能够使用的 GPU
torch.cuda.is_available()
查看本机 GPU 数量
import torch
n_gpu = torch.cuda.device_count()
print(n_gpu)
将模型部署到 GPU
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
扫码打赏,你说多少就多少