PyTorch 文章

PyTorch GPU 配置指南:CUDA 安装与环境验证
2024-02-09 Raikay

PyTorch GPU 配置指南:CUDA 安装与环境验证

PyTorch 作为主流的深度学习框架,在安装时严格区分了 CPU 版本与 GPU 版本。这两个版本对应不同的底层二进制分发包,并非仅通过代码参数进行区分。在实际开发与部署中,准确判断当前环境并进行相 …

PyTorch CUDA GPU加速
阅读更多
结合代码,看懂PyTorch的自动微分(Autograd)
2023-12-16 Raikay

结合代码,看懂PyTorch的自动微分(Autograd)

在深度学习中,PyTorch 的自动微分(Autograd)机制极大地简化了模型训练过程。在实际的训练代码中,自动微分并不是一个孤立的函数,而是通过一套标准的训练闭环来体现的。 …

PyTorch 自动微分 Autograd
阅读更多
四维张量计算原理
2023-10-22 Raikay

四维张量计算原理

四维张量 sum(dim) 计算原理 示例张量 t4,shape = (2, 2, 2, 3) → 2个组 × 2个块 × 2行 × 3列 📦 原始张量结构 第 0 组 (dim=0 索引0) …

张量 PyTorch 深度学习
阅读更多