![](https://img-blog.csdnimg.cn/20190927151053287.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
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 · 1002 阅读 · 6 评论 -
pytorch入坑指南
1. 内存爆炸问题1.1 loss.item或 float(loss)首先附上主要代码下图接上图上面为定义的TextCNN模型,下图为主要的训练及预测测试集的过程下图接上图可以从第三张图片看到绿框部分有一个tr_loss = float(tr_loss)。在pytorch中涉及需要求导/梯度变量的计算将保存在内存中,如果不释放指向计算图形的指针,这些变量在循环训练中就会超出你内...原创 2020-03-25 10:26:02 · 1259 阅读 · 7 评论 -
pytorch基础
import torchimport 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 · 693 阅读 · 3 评论 -
pytorch 安装
首先进入pytorch官网—Get Started—选择环境、版本等复制下方的代码windows环境下,选择开始,找到cmd,右键使用管理员权限打开,复制刚才的代码,安装即可ps:如果不使用管理员权限,可能会出现Missing Write Permission 或者 http error等问题,但是使用管理员权限后,发现两个问题都解决了。(本人电脑在装完conda后就设置为清华镜像源,如...原创 2019-04-09 09:54:39 · 307 阅读 · 3 评论 -
torch.nn.functional & torch.nn.init
functional主要是各种函数,包含卷积网络函数、线性函数、激活函数、距离函数等,以下为思维导图:原创 2019-04-28 09:52:57 · 285 阅读 · 5 评论