14 lines
386 B
Python
14 lines
386 B
Python
#!/usr/bin/env python3
|
|
import torch
|
|
import safetensors
|
|
import transformers
|
|
|
|
print(f"torch version: {torch.__version__}")
|
|
print(f"safetensors version: {safetensors.__version__}")
|
|
print(f"transformers version: {transformers.__version__}")
|
|
print(f"CUDA available: {torch.cuda.is_available()}")
|
|
|
|
x = torch.tensor([1.0, 2.0, 3.0])
|
|
y = torch.tensor([4.0, 5.0, 6.0])
|
|
print(f"x + y = {x + y}")
|