7 lines
304 B
Python
7 lines
304 B
Python
import torch
|
|
# Check the PyTorch version
|
|
print("Torch Version:", torch.__version__)
|
|
# Check if PyTorch has CUDA support enabled (should be True)
|
|
print("CUDA Available:", torch.cuda.is_available())
|
|
# Print the CUDA version PyTorch was compiled with
|
|
print("CUDA Version used by Torch:", torch.version.cuda) |