device_management: CUDA Device Management

In wrathematics/cur: 'CUDA' Interface for R

Description

CUDA device management utilities.

Usage

cudaDeviceReset()

cudaGetDevice()

cudaGetDeviceCount()

cudaSetDevice(device)

Arguments

device

A non-negative integer corresponding to the GPU you want to use.

Details

cudaDeviceReset() Destroys all allocations and resets state on the current device in the current process.

cudaGetDevice() returns the device number (0, 1, ...) currently in use.

cudaGetDeviceCount() returns the number of available GPU's.

cudaSetDevice() sets the device to the supplied number (0, 1, ...).

References

NVIDIA CUDA Runtime API https://docs.nvidia.com/cuda/cuda-runtime-api/index.html

wrathematics/cur documentation built on May 21, 2019, 10:14 a.m.