CNN2 / check_libs.py
d-e-e-k-11's picture
Upload folder using huggingface_hub
294928d verified
raw
history blame contribute delete
681 Bytes
import sys
print(f"Python version: {sys.version}")
try:
import tensorflow as tf
print(f"TensorFlow version: {tf.__version__}")
except Exception as e:
print(f"TensorFlow import failed: {e}")
try:
import keras
print(f"Keras version: {keras.__version__}")
except Exception as e:
print(f"Keras import failed: {e}")
try:
from tensorflow import keras as tf_keras
print("from tensorflow import keras successful")
except Exception as e:
print(f"from tensorflow import keras failed: {e}")
try:
import torch
print(f"PyTorch version: {torch.__version__}")
except Exception as e:
print(f"PyTorch import failed: {e}")