a0a7's picture
Upload 8 files
3663dd0 verified
raw
history blame contribute delete
65 Bytes
import torch
print("CUDA available:", torch.cuda.is_available())