部分引用自 https://zhuanlan.zhihu.com/p/59205847
首先需要导入如下库
import collections
import os
import shutil
import tqdm
import numpy as np
import PIL.Image
import torch
import torchvision
1 基础配置
检查pytorch版本以及cuda
torch.__version__
torch.version.cuda
torch.backends.cudnn.version()
torch.cuda.get_device_name(0)
固定随机种子
torch.manual_seed(0)
torch.cuda.manual_seed_all(0)
