点赞收藏:PyTorch常用代码段整理合集

  1. 云栖社区>
  2. 机器学习算法与Python学习>
  3. 博客>
  4. 正文

点赞收藏:PyTorch常用代码段整理合集

初商 2019-09-02 23:42:37 浏览1038
展开阅读全文

机器之心转载

来源:知乎

作者:张皓


本文代码基于 PyTorch 1.0 版本,需要用到以下包

import collections
import os
import shutil
import tqdm

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

基础配置

检查 PyTorch 版本

torch.__version__               # PyTorch version
torch.version.cuda              # Corresponding CUDA version
torch.backends.cudnn.version()  # Corresponding cuDNN version
torch.cuda.g

网友评论

登录后评论
0/500
评论
初商
+ 关注