腾讯云开发者社区-腾讯云

开发者社区

文档建议反馈控制台

最新优惠活动

文章/答案/技术大牛

发布

AI工程落地

AI不止训练，专注工程落地

专栏作者

12

文章

11752

阅读量

17

订阅数

大语言模型--Tensor Parallel（张量并行）

pytorch huggingface-transformers nvidia LLM

NCCL是一个Nvidia专门为多GPU之间提供通讯的通讯库，或者说是一个多GPU卡通讯的框架，提供了包括AllReduce、Broadcast、Reduce、AllGather、ReduceScatter等集合通讯API。NCCL屏蔽了底层复杂的细节，向上提供API供训练框架调用，向下连接机内机间的GPU以完成模型参数的高效传输。

2024-01-11

7066

置顶

大语言模型--开源数据集

dataset huggingface-transformers LLM

Huggingface开源大模型排行榜: Open LLM Leaderboard - a Hugging Face Space by HuggingFaceH4

2023-12-22

4140

TensorRT LLM--RMSNorm

2023-11-22

1400

TensorRT LLM vs OpenPPL LLM

PPL LLM只支持baichuan、chatglm、llama三个模型，Tensor-LLM支持几乎所有大模型。

2023-11-21

6210

TensorRT LLM--Beam Search

Beam Search思想介绍：如何通俗的理解beam search？ - 知乎 (zhihu.com)

2023-11-16

3510

TensorRT LLM--Paged KV Cache

技术出处：vLLM: Easy, Fast, and Cheap LLM Serving with PagedAttention | vLLM Blog

2023-11-16

8790

TensorRT LLM--In-Flight Batching

TensorRT LLM依赖于一个名为Batch Manager的组件来支持请求的即时批处理，该技术旨在减少队列中的等待时间，达到更高的GPU利用率。

2023-11-16

9010

Mxnet模型性能优化

导语：模型优化有很多方法，比如模型压缩、模型剪纸、转tensorrt等。本文讨论mxnet模型tesorrt优化，主要解决动态batch、Op不支持问题。

2022-02-25

4950

记一次排查模型推理变慢原因

导语：使用nvidia dali库来做图片加载和transforms得到的tensor，比pillow+torchvision得到的tensor，在模型推理时候慢三倍。对比tensor，虽然存在精度上的损失，但没发现其他任何问题，最终定位出来是显存中tensor不连续。

2022-02-25

2.1K0

不重训模型情况下，MMDetection模型V1升级V2

机器学习神经网络深度学习人工智能 github

导语：MMDetection是一个基于Pytorch实现的目标检测框架，支持Faster-RCNN、Mask-RCNN、Fast-RCNN等主流目标检测算法。旧版本的MMDetection会有很多限制，比如不能运行在新的cuda上、做模型优化也受到很多限制。MMDetection从v1到v2改变很大，从权重到模型申明语句都经历了翻天覆地的变化。本文讨论在不重新训练模型的情况下，把mmdet升级到2.x的办法。

2022-02-25

6770

Golang分布式限流开源实现

导语：流量上涨常常造成系统的不稳定，进而出现雪崩。本文讨论常见的限流算法，以及对比一些开源实现。

2022-02-25

2K0

Pytorch转TensorRT实践

导语：TensorRT立项之初的名字叫做GPU Inference Engine(简称GIE)，是NVIDIA 推出的一款基于CUDA和cudnn的神经网络推断加速引擎，TensorRT现已支持TensorFlow、Caffe、Mxnet、Pytorch等几乎所有的深度学习框架，将TensorRT和NVIDIA的GPU结合起来，能在几乎所有的框架中进行快速和高效的部署推理。

2022-02-25

2.5K1

没有更多了

社区活动

腾讯技术创作狂欢月

“码”上创作 21 天，分 10000 元奖品池！

Python精品学习库

代码在线跑，知识轻松学

博客搬家 | 分享价值百万资源包

自行/邀约他人一键搬运博客，速成社区影响力并领取好礼

技术创作特训营·精选知识专栏

往期视频·千货材料·成员作品最新动态