Update documents

Author: Eleven1Liu

README.md

@@ -10,8 +10,8 @@ This is an on-going development so many improvements are still being made. Comme
 
 ## Environments
 - Python: 3.8+
-- CUDA: 11.8 (if training neural networks by GPU)
-- Pytorch: 2.1.2
+- CUDA: 11.8, 12.1 (if training neural networks by GPU)
+- Pytorch: 2.0.1+
 
 If you have a different version of CUDA, follow the installation instructions for PyTorch LTS at their [website](https://pytorch.org/).
 

docs/cli/ov_data_format.rst

@@ -25,8 +25,8 @@ Install LibMultiLabel from Source
 * Environment
 
     * Python: 3.8+
-    * CUDA: 11.8 (if training neural networks by GPU)
-    * Pytorch 2.1.2
+    * CUDA: 11.8, 12.1 (if training neural networks by GPU)
+    * Pytorch 2.0.1+
 
 It is optional but highly recommended to
 create a virtual environment.

setup.cfg

@@ -12,6 +12,7 @@ project_urls =
     Documentation = https://www.csie.ntu.edu.tw/~cjlin/libmultilabel
     Source Code = https://github.com/ASUS-AICS/LibMultiLabel/
 classifiers =
+    Environment :: GPU :: NVIDIA CUDA :: 12.1
     Environment :: GPU :: NVIDIA CUDA :: 11.8
     Intended Audience :: Developers
     Intended Audience :: Education