weixin_41596463的博客

1、问题 安装好torch gpu版本之后，torch.cuda.is_available()一直返回False; 但是执行 torch.backends.cudnn.enabled is TRUE. 执行Nvidia-smi 命令没有报错，能够显示驱动信息； 在网上，搜的解决方法：执行命令： sudo apt-get install nvidia-cuda-toolkit 仍然...

# 1、创建tensor t.Tensor(*size) # 基础构造函数 t.ones(*size) # 全为1的tensor t.zeros(*size) # 全为0的tensor t.eye(*size) # 单位矩阵 t.arange(s, ...

1、全连接 import torch as t from torch.autograd import Variable as V from torch import nn class Linear(nn.Module): def __init__(self, in_features, out_features): super(Linear, self).__init...

1、数据加载 数据集被抽象为Dataset类, 只负责数据的抽象 两个魔法方法： __getitem__:返回一条数据或一个样本， __len__：返回样本的数量 ImageFolder() 对数据进行shuffle和并行加速 DataLoader(dataset, batch_size=1, shuffle=False,...) 2、计算机视觉工具包：torchvision...