546 B
546 B
Environment Checklist
- PyTorch imports successfully
torch.cuda.is_available()isTrue- At least one CUDA device is visible
- The GPU name matches the machine you expect to be using
- Device capability is printed and recorded
- Triton imports successfully, or you know why it does not
torch.version.cudais visible when using CUDA-enabled PyTorchnvcc --versionworks if you plan to build the CUDA extensionnvidia-smiworks if the driver stack is installed
If any line above fails, fix that before working on later tasks.