pytorch
weixin_41596463
这个作者很懒,什么都没留下…
展开
-
torch.cuda.is_available()返回false的解决办法
1、问题安装好torch gpu版本之后,torch.cuda.is_available()一直返回False; 但是执行 torch.backends.cudnn.enabled is TRUE.执行Nvidia-smi 命令没有报错,能够显示驱动信息;在网上,搜的解决方法:执行命令:sudo apt-get install nvidia-cuda-toolkit仍然...原创 2019-10-12 18:29:10 · 8557 阅读 · 1 评论 -
pytorch入门
# 1、创建tensort.Tensor(*size) # 基础构造函数t.ones(*size) # 全为1的tensort.zeros(*size) # 全为0的tensort.eye(*size) # 单位矩阵t.arange(s, ...原创 2019-08-30 20:28:49 · 96 阅读 · 0 评论 -
pytorch神经网络工具箱nn
1、全连接import torch as tfrom torch.autograd import Variable as Vfrom torch import nnclass Linear(nn.Module): def __init__(self, in_features, out_features): super(Linear, self).__init...原创 2019-08-31 10:52:32 · 257 阅读 · 0 评论 -
pytorch常用的工具
1、数据加载数据集被抽象为Dataset类, 只负责数据的抽象两个魔法方法:__getitem__:返回一条数据或一个样本,__len__:返回样本的数量ImageFolder()对数据进行shuffle和并行加速DataLoader(dataset, batch_size=1, shuffle=False,...)2、计算机视觉工具包:torchvision...原创 2019-08-31 11:23:23 · 498 阅读 · 0 评论