OneFlow documentation

OneFlow documentation

Troubleshooting

OneFlow Python API

oneflow
oneflow.nn
oneflow.nn.functional
oneflow.Tensor
Tensor Attributes
oneflow.autograd
oneflow.cuda
oneflow.distributed
oneflow.linalg
oneflow.nn.init
oneflow.optim
oneflow.nn.Graph
oneflow.nn.image
oneflow.utils.data
oneflow.one_embedding
Environment Variables

oneflow.cuda.is_available¶

oneflow.cuda.is_available() → bool ¶: Returns a bool indicating if CUDA is currently available.

oneflow.cuda.device_count

Copyright © 2020, OneFlow | Built with Sphinx and @pradyunsg's Furo theme. | Show Source