torch.cuda.gds.is_available

torch.cuda.gds.is_available()

Return True if GPUDirect Storage (GDS) support is built in.

This requires a Linux build with USE_CUFILE enabled (the default) and with cuFile (CUDA) or hipFile (ROCm) available at build time. See hipFile (GPUDirect Storage) for the ROCm requirements.

Return type:

bool