init commit
This commit is contained in:
13
test.py
Normal file
13
test.py
Normal file
@@ -0,0 +1,13 @@
|
||||
#!/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}")
|
||||
Reference in New Issue
Block a user