paddle.cuda
paddle.cuda 目录下包含飞桨框架支持的 torch.cuda 兼容函数与模块接口
PyTorch 兼容函数
API 名称 |
API 功能 |
|---|---|
|
|
检测 CUDA 错误码 |
|
|
以模块的形式返回 CUDA Runtime 对象 |
|
|
获取当前 CUDA 流 |
|
|
获取 CUDA 设备属性 |
|
|
获取随机数生成器状态 |
|
|
检查 CUDA 是否可用 |
|
|
判断 CUDA 是否已经初始化 |
|
|
设置全局随机种子 |
|
|
获取指定设备上的全局空闲显存和显存总量 |
|
|
设置随机数生成器状态 |
|
|
同步 CUDA 设备 |
|
|
返回当前设备的索引 |
|
|
返回可用的 CUDA 设备数量 |
|
|
释放当前设备上所有未占用的缓存内存 |
|
|
返回当前设备上分配的内存总量 |
|
|
返回当前设备上由缓存分配器管理的内存总量 |
|
|
设置当前设备 |
|
|
CUDA 流类 |
|
|
从外部流创建 Paddle 流 |
|
|
临时选择设备使用 |
|
|
设置设备随机种子 |
|
|
获取最大内存分配量 |
|
|
重置峰值内存统计 |
|
|
返回指定设备的计算能力 |