Sharfraz's picture
Create test pytorch
f35f8d1 verified
raw
history blame contribute delete
90 Bytes
import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())