新手入门
用户指南
API 参考
开发者指南
论坛
官网
GitHub
Python API
数据(Data)
张量(Tensor)
函数式(Functional)
模块式(Module)
自动微分(Auto-diff)
优化器(Optimizer)
即时编译(JIT)
分布式训练(Distributed)
量化(Quantization)
序列化(Serialization)
随机(Random)
模型中心(Hub)
日志(Logger)
设备(Device)
megengine.is_cuda_available
megengine.get_device_count
megengine.get_default_device
megengine.set_default_device
megengine.get_mem_status_bytes
megengine.set_prealloc_config
查询版本信息
Utils 模块
Edit this page
megengine.is_cuda_available
¶
is_cuda_available
(
)
[源代码]
¶
cuda设备是否在当前机器可用。
返回类型
bool
设备(Device)
megengine.get_device_count