Enum tch::Cuda [−][src]
pub enum Cuda {}
Expand description
Cuda related helper functions.
Implementations
Returns the number of GPU that can be used.
Returns true if cuda support is available.
Returns true if cudnn support is available.
Returns true if cudnn is enabled by the user.
This does not indicate whether cudnn is actually usable.
Enable or disable cudnn.
Sets cudnn benchmark mode.
When set cudnn will try to optimize the generators durning the first network runs and then use the optimized architecture in the following runs. This can result in significant performance improvements.