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)
|