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

初商 2019-09-02

机器学习算法与python学习

机器之心转载

来源:知乎

作者:张皓


本文代码基于 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
登录 后评论
下一篇
corcosa
13015人浏览
2019-10-08
相关推荐
云栖Android精华文章合集
3538人浏览
2017-06-03 20:23:25
301转向代码合集
532人浏览
2017-06-25 17:01:00
写技术博客那点事
1146人浏览
2019-02-15 10:02:52
0
0
0
1057