dreambooth-for-diffusion/tools/test_cuda.py

import torch
print(torch.cuda.is_available())