美文网首页
Pytorch常用集合

Pytorch常用集合

作者: LaLa_2539 | 来源:发表于2019-03-30 17:24 被阅读0次

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

相关文章

网友评论

      本文标题:Pytorch常用集合

      本文链接:https://www.haomeiwen.com/subject/jddsbqtx.html