From 2bf117cf71fb4ddb539bf36807ea0af1f4a67edc Mon Sep 17 00:00:00 2001
From: Cheng Lai <laicheng_VIP@163.com>
Date: Fri, 3 Feb 2023 16:00:57 +0800
Subject: [PATCH] TensorLayerX 0.5.8 Release (#68)

---
 README.md                    |  5 +++++
 README.rst                   |  2 +-
 setup.py                     |  2 +-
 tensorlayerx/__init__.py     | 13 ++++++++++++-
 tensorlayerx/package_info.py |  2 +-
 5 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 672055b..3601a20 100644
--- a/README.md
+++ b/README.md
@@ -66,6 +66,11 @@ More resources can be found [here](https://github.com/tensorlayer)
 
 ## Installation
 
+### TensorLayerX corresponds to the back-end version
+| TensorLayerX | TensorFlow | MindSpore | PaddlePaddle|PyTorch|
+| :-----:| :----: | :----: |:-----:|:----:|
+|  v0.5.8  | v2.4.0 | v1.8.1 | v2.2.0 | v1.10.0 |
+| v0.5.7 | v2.0.0 | v1.6.1 | v2.0.2 | v1.10.0 |
 ### Via docker
 
 Docker is an open source application container engine. In the [TensorLayerX Docker Repository](https://hub.docker.com/repository/docker/tensorlayer/tensorlayerx), 
diff --git a/README.rst b/README.rst
index 716e1cf..ff2ba3a 100644
--- a/README.rst
+++ b/README.rst
@@ -89,5 +89,5 @@ License
 
 TensorLayerX is released under the Apache 2.0 license.
 
-.. |TENSORLAYER-LOGO| image:: https://git.openi.org.cn/TensorLayer/tensorlayer3.0/src/branch/master/img/tl_transparent_logo.png
+.. |TENSORLAYER-LOGO| image:: https://git.openi.org.cn/hanjr/tensorlayerx-image/raw/branch/master/tlx-LOGO--02.jpg
    :target: https://tensorlayerx.readthedocs.io/en/latest/
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 9d5f0a4..8f4ea19 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@
 
 MAJOR = 0
 MINOR = 5
-PATCH = 7
+PATCH = 8
 PRE_RELEASE = ''
 # Use the following formatting: (major, minor, patch, prerelease)
 VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE)
diff --git a/tensorlayerx/__init__.py b/tensorlayerx/__init__.py
index a2e6cf6..1970a1a 100644
--- a/tensorlayerx/__init__.py
+++ b/tensorlayerx/__init__.py
@@ -4,7 +4,7 @@
 
 # import backend
 from .backend import *
-
+import warnings
 import os
 
 from tensorlayerx.package_info import (
@@ -33,3 +33,14 @@
     # global vars
     global_flag = {}
     global_dict = {}
+
+backend_v = {
+    'tensorflow': '2.4.0',
+    'mindspore': '1.8.1',
+    'paddle': '2.2.0',
+    'torch': '1.10.0',
+}
+
+if BACKEND_VERSION != backend_v[BACKEND]:
+    warnings.warn("The version of the backend you have installed does not match the specified backend version "
+                  "and may not work, please install version {} {}.".format(BACKEND, backend_v[BACKEND]))
\ No newline at end of file
diff --git a/tensorlayerx/package_info.py b/tensorlayerx/package_info.py
index a07fa2c..80d5442 100644
--- a/tensorlayerx/package_info.py
+++ b/tensorlayerx/package_info.py
@@ -4,7 +4,7 @@
 
 MAJOR = 0
 MINOR = 5
-PATCH = 7
+PATCH = 8
 PRE_RELEASE = ''
 # Use the following formatting: (major, minor, patch, prerelease)
 VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE)