algorithm_platform/models_processing/model_koi/test.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @FileName  :test.py
# @Time      :2025/3/5 18:11
# @Author    :David
# @Company: shenyang JY

import tensorflow as tf
print("TensorFlow 版本:", tf.__version__)
print("GPU 是否可用:", tf.config.list_physical_devices('GPU'))