Pytorch
文章平均质量分 67
三个刺客
这个作者很懒,什么都没留下…
展开
-
Jetson nano安装torch和torchvision
Jetson nano安装torch和torchvision原创 2022-08-25 21:01:48 · 3408 阅读 · 1 评论 -
pytorch train.py与test.py代码流程
首先讲解train.py,包括以下步骤1、配置训练集与测试集train_data = torchvision.datasets.CIFAR10(root="./dataset", train=True, transform=torchvision.transforms.ToTensor(), download=True)test_data = torchvision.datasets.CIFAR10(root="原创 2022-04-28 16:40:25 · 5269 阅读 · 0 评论 -
Pytorch搭建网络模型
这篇博文是参考小土堆的Pytorch的入门视频,主要是通过查询文档的方式讲解如何搭建卷积神经网络,讲解的主要是Pytorch核心包的TORCH.NN中的内容(nn是Neural Network的缩写)通常我们定义好的网络模型会继承torch.nn.Module类,该类为我们定义好了神经网络骨架。卷积层 Convolution Layers对于图像处理来说,我们通常使用二维卷积,即使用torch.nn.Conv2d类:创建该类时,我们通常只需要传入以下几个参数,其他不常用...原创 2022-04-28 15:28:34 · 2469 阅读 · 0 评论 -
Pytorch中实现CPU和GPU之间的切换
如何在pytorch中指定CPU和GPU进行训练,以及cpu和gpu之间切换由CPU切换到GPU,要修改的几个地方:网络模型、损失函数、数据(输入,标注)# 创建网络模型tudui = Tudui()if torch.cuda.is_available(): tudui = tudui.cuda()# 损失函数loss_fn = nn.CrossEntropyLoss()if torch.cuda.is_available(): loss_fn = loss_fn.c原创 2022-04-27 22:11:31 · 12792 阅读 · 0 评论 -
pytorch安装方法
进入pytorch官网,链接如下Start Locally | PyTorch安装10.2版本在终端中输入conda install pytorch torchvision torchaudio cudatoolkit=10.2此时提示报错,需要换源换源方法如下:打开.condarc 我的位置在C:\Users\mxh ,自行查看自己电脑的位置然后把下面源粘贴到.condarc文件中,并覆盖掉channels: - http://mirrors.tun...原创 2022-03-07 11:33:43 · 762 阅读 · 0 评论