File tree Expand file tree Collapse file tree 3 files changed +8
-8
lines changed Expand file tree Collapse file tree 3 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -10,8 +10,8 @@ This is an on-going development so many improvements are still being made. Comme
10
10
11
11
## Environments
12
12
- Python: 3.7+
13
- - CUDA: 10.2 (if training neural networks by GPU)
14
- - Pytorch 1.12 +
13
+ - CUDA: 11.6 (if training neural networks by GPU)
14
+ - Pytorch 1.13.1 +
15
15
16
16
If you have a different version of CUDA, follow the installation instructions for PyTorch LTS at their [ website] ( https://pytorch.org/ ) .
17
17
Original file line number Diff line number Diff line change @@ -25,8 +25,8 @@ Install LibMultiLabel from Source
25
25
* Environment
26
26
27
27
* Python: 3.7+
28
- * CUDA: 10.2 (if training neural networks by GPU)
29
- * Pytorch 1.12 +
28
+ * CUDA: 11.6 (if training neural networks by GPU)
29
+ * Pytorch 1.13.1 +
30
30
31
31
It is optional but highly recommended to
32
32
create a virtual environment.
Original file line number Diff line number Diff line change 1
1
[metadata]
2
2
name = libmultilabel
3
- version = 0.1.6
3
+ version = 0.1.7
4
4
author = LibMultiLabel Team
5
5
license = MIT License
6
6
license_file = LICENSE
@@ -12,7 +12,7 @@ project_urls =
12
12
Documentation = https://www.csie.ntu.edu.tw/~cjlin/libmultilabel
13
13
Source Code = https://github.com/ASUS-AICS/LibMultiLabel/
14
14
classifiers =
15
- Environment :: GPU :: NVIDIA CUDA :: 10.2
15
+ Environment :: GPU :: NVIDIA CUDA :: 11.6
16
16
Intended Audience :: Developers
17
17
Intended Audience :: Education
18
18
Intended Audience :: Science/Research
@@ -29,8 +29,8 @@ install_requires =
29
29
pandas
30
30
PyYAML
31
31
scikit-learn
32
- torch>=1.12.0
33
- torchmetrics>=0.9.2
32
+ torch>=1.13.1
33
+ torchmetrics ==0.10.3
34
34
torchtext>=0.13.0
35
35
pytorch-lightning ==1.7.7
36
36
tqdm
You can’t perform that action at this time.
0 commit comments