部分引用自 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)