Pytorch
王大银子
这个作者很懒,什么都没留下…
展开
-
代码记录(4)
【pytorch】self.modules() 和 self.children()的区别 self.modue和self.children的区别与联系 import torch from torch import nn # hyper parameters in_dim=1 n_hidden_1=1 n_hidden_2=1 out_dim=1 class Net(nn.Module): def __init__(self, in_dim, n_hidden_1, n_hidden_2,原创 2020-12-29 22:38:09 · 845 阅读 · 0 评论 -
代码记录(3)
torch.mul 点乘都是broadcast的,可以用torch.mul(a, b)实现,也可以直接用*实现。 >>> a = torch.ones(3,4) >>> a tensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]]) >>> b = torch.Tensor([1,2,3]).reshape((3,1)) >>> b原创 2020-12-09 20:19:38 · 301 阅读 · 0 评论 -
Linux下Anaconda中指定的环境中安装Pytorch-GPU
1.硬件方面首先是检查电脑(服务器)上是否安装了cudnn,cuda。 在安装了的前提下,检查cudnn,以及cuda的版本号: 查看CUDA版本: cat /usr/local/cuda/version.txt 查看cudnn版本: cat /usr/local/cuda/include/include/cudnn.h | grep CUDNN_MAJOR -A 2 2.查看torch不同版本发布日期: https://mirrors.tuna.tsinghua.edu.cn..原创 2020-11-03 17:24:09 · 740 阅读 · 0 评论