
AI-Pytorch使用
pytorch框架
爱编程的喵喵
双985硕士毕业,现担任AI大模型全栈工程师一职,热衷于将数据思维应用到工作与生活中。曾在阿里云、科大讯飞、CCF等比赛获得多次Top名次。官方认证“人工智能领域优质创作者”,入选2023年中国开发者影响力年度榜单,荣获付费内容优质创作者称号。代表专栏《Python基础课程》、《Linux解决方案》、《Windows实用技巧》、《机器学习理论与实战》、《从零开始学Java》等。
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
如何安装pytorch
pytorch的安装,打开官网即有相应说明,但是,根据官网的方式,安装特别慢,因此可以下载whl文件,pip install安装。whl文件的网址:打开pytorch官网就,找到previous pytroch versions 然后可以找到via pip 假如我要安装cuda9.2的版本,可以点击倒数第二个链接,进入如下界面: 可以看到,这里有适用cuda92的不同pytorch版本,不同...原创 2020-04-13 12:52:03 · 1076 阅读 · 6 评论 -
pytorch入坑指南
1. 内存爆炸问题 1.1 loss.item或 float(loss) 首先附上主要代码 下图接上图 上面为定义的TextCNN模型,下图为主要的训练及预测测试集的过程 下图接上图 可以从第三张图片看到绿框部分有一个tr_loss = float(tr_loss)。在pytorch中涉及需要求导/梯度变量的计算将保存在内存中,如果不释放指向计算图形的指针,这些变量在循环训练中就会超出你内...原创 2020-03-25 10:26:02 · 1367 阅读 · 7 评论 -
pytorch基础
import torch import numpy as np %config ZMQInteractiveShell.ast_node_interactivity = 'all' # pytorch中tensor的类型:主要分为两类 # FloatTensor and IntTensor # 设置默认的tensor类型 # torch.set_default_tensor_type(tor...原创 2019-04-10 16:57:14 · 791 阅读 · 3 评论 -
pytorch 安装
首先进入pytorch官网—Get Started—选择环境、版本等复制下方的代码 windows环境下,选择开始,找到cmd,右键使用管理员权限打开,复制刚才的代码,安装即可 ps:如果不使用管理员权限,可能会出现Missing Write Permission 或者 http error等问题,但是使用管理员权限后,发现两个问题都解决了。(本人电脑在装完conda后就设置为清华镜像源,如...原创 2019-04-09 09:54:39 · 362 阅读 · 3 评论 -
torch.nn.functional & torch.nn.init
functional主要是各种函数,包含卷积网络函数、线性函数、激活函数、距离函数等,以下为思维导图:原创 2019-04-28 09:52:57 · 324 阅读 · 5 评论