部分引用自 https://zhuanlan.zhihu.com/p/59205847

首先需要導(dǎo)入如下庫

import collections
import os
import shutil
import tqdm

import numpy as np
import PIL.Image
import torch
import torchvision

1 基礎(chǔ)配置

檢查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)