PYTORCH文档
PyTorch是一个使用GPU和CPU进行深度学习的优化张量库。
本文档中描述的功能按版本状态分类:
稳定:这些功能将被长期维护,并且在文档中通常不应该有重大的性能限制或缺口。我们还希望保持向后兼容性(虽然突破性的变化可能会发生,通知将提前一个版本)。
测试版:这些功能被标记为测试版,因为API可能会根据用户反馈而改变,因为性能需要改进,或者因为跨运营商的覆盖范围尚未完成。对于测试版特性,我们承诺将该特性进行稳定分类。然而,我们并不承诺向后兼容。
原型:这些特性通常不作为PyPI或Conda等二进制发行版的一部分提供,除了有时在运行时标志后面,并且处于反馈和测试的早期阶段。
社区
开发者笔记
- CUDA自动混合精度示例
- 亲笔签名的力学
- 广播语义学
- CPU线程和TorchScript推断
- CUDA语义
- 分布式数据并行
- 扩展PyTorch
- 用亲笔签名扩展torch.func。功能
- 常见问题
- Gradcheck力学
- HIP (ROCm)语义
- 适用于大规模部署的功能
- 模块
- MPS后端
- 多重处理最佳实践
- 数字准确度
- 再现性
- 序列化语义
- Windows常见问题
火炬.编译
语言绑定
Python API
- 火炬
- torch.nn
- 火炬. nn .功能
- 火炬。张量
- 张量属性
- 张量视图
- 手电筒
- 火炬,亲笔签名
- 火炬.图书馆
- 火炬. cuda
- 流上下文
- torch . cuda . can _设备_访问_对等
- torch.cuda.current_blas_handle
- 火炬. cuda.current_device
- 火炬. cuda.current_stream
- 火炬. cuda.default_stream
- 设备
- 火炬. cuda .设备_计数
- 设备_of
- torch.cuda.get_arch_list
- torch . cuda . get _ device _ capacity
- torch.cuda.get_device_name
- torch . cuda . get _设备_属性
- torch.cuda.get_gencode_flags
- 火炬. cuda.get_sync_debug_mode
- 火炬. cuda.init
- torch . cuda . IPC _收集
- torch . cuda . is _可用
- torch.cuda.is_initialized
- torch.cuda.memory _ usage用法
- torch . cuda . set _设备
- torch.cuda.set_stream
- 火炬. cuda.set_sync_debug_mode
- 火炬. cuda.stream
- torch.cuda.synchronize
- 火炬. cuda .利用
- 火炬. cuda .温度
- torch.cuda.power_draw
- 火炬. cuda.clock_rate
- torch . cuda . out of memory错误
- 随机数发生器
- 通信集体
- 流和事件
- 图表(测试版)
- 内存管理
- NVIDIA工具扩展(NVTX)
- Jiterator (beta)
- 蒸汽消毒剂(原型)
- 火炬. mps
- 火炬.后端
- 火炬.分布式
- 火炬.分布式.算法.加入
- 火炬.分布式.弹性
- 火炬.分布式. fsdp
- 火炬.分布式. optim
- 火炬.分布式.张量.并行
- 火炬.分布式.检查点
- 火炬.分发
- 火炬。_发电机
- torch.fft
- 火炬.功能
- 火炬未来
- torch.fx
- 火炬中心
- torch.jit
- torch.linalg
- 火炬.监视器
- 火炬信号
- 火炬.特别
- 火炬.覆盖
- 火炬.包装
- torch.profiler
- 火炬. nn.init
- torch.onnx
- torch.onnx诊断
- 火炬. optim
- 复数
- DDP通信挂钩
- 流水线并行性
- 量化
- 分布式RPC框架
- 火炬.随机
- 火炬,蒙面
- 火炬.嵌套
- 火炬。稀疏
- 火炬。储存;储备
- 火炬测试
- torch.utils .基准
- torch.utils .瓶颈
- torch.utils.checkpoint
- torch.utils.cpp_extension
- torch.utils.data
- torch.utils.jit
- torch.utils.dlpack
- torch.utils.mobile_optimizer
- torch.utils.model_zoo
- torch.utils.tensorboard
- 类型信息
- 名为张量
- 命名张量算子覆盖
- 火炬。__配置_ _
图书馆