Skip to content

Commit

Permalink
Update requirements.txt
Browse files Browse the repository at this point in the history
  • Loading branch information
Puiching-Memory committed Dec 22, 2024
1 parent 4e81b50 commit 895f448
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,17 +4,17 @@
torchinfo==1.8.0
opencv-python==4.10.0.84
pillow==10.4.0
psutil==6.1.0
psutil==6.1.1
matplotlib==3.10.0
pyecharts==2.0.7
open3d==0.18.0 # https://www.open3d.org/docs/latest/getting_started.html
#grad-cam==1.5.4
albumentations==1.4.22
albumentations==1.4.23

# log tools (must be installed)
loguru==0.7.3
rich==13.9.4
swanlab==0.4.0
swanlab==0.4.1
#tqdm==4.67.1

# export tools (optional)
Expand All @@ -25,6 +25,8 @@ onnxsim==0.4.36
torch-tensorrt==2.5.0 # https://pytorch.org/TensorRT/getting_started/installation.html#installation
tensorrt==10.3.0
# pip install "nvidia-modelopt[all]~=0.19.0" --extra-index-url https://pypi.nvidia.com # https://nvidia.github.io/TensorRT-Model-Optimizer/getting_started/2_installation.html
pnnx==20240819
ncnn==1.0.20240820
monkeytype==23.3.0

# datasets tools (optional)
Expand All @@ -40,7 +42,7 @@ Sophia-Optimizer==0.2.5
optuna==4.1.0

# doc tools (optional)
mkdocs-material[imaging]==9.5.48
mkdocs-material[imaging]==9.5.49
mkdocs-git-revision-date-localized-plugin==1.3.0
mkdocs-git-authors-plugin==0.9.2
mkdocs-static-i18n==1.2.3
Expand Down

0 comments on commit 895f448

Please sign in to comment.