1580158
/
PBRemTools
mirror of https://github.com/mattyamonaca/PBRemTools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #34 from udon-universe/feature/pin-onnxrutime-gpu-version-and-etc 

update package list with dict and pin
pull/38/head
mattya_monaca 2023-05-22 11:10:50 +09:00 committed by GitHub
parent 302252a7ca efe6752004
commit b179be6192
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
install.py
View File

@ -1,19 +1,19 @@
import launch
packages = [
"onnx",
"onnxruntime-gpu",
"opencv-python",
"numpy",
"Pillow",
"segmentation-refinement",
"scikit-learn",
"clip",
]
packages = {
"onnx": "onnx",
"onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
"opencv-python": "opencv-python",
"numpy": "numpy",
"Pillow": "Pillow",
"segmentation-refinement": "segmentation-refinement",
"scikit-learn": "scikit-learn",
"clip": "clip",
}
for package in packages:
if not launch.is_installed(package):
launch.run_pip(f'install {package}', desc=f'{package} for PBRemTools')
for name, target in packages.items():
if not launch.is_installed(name):
launch.run_pip(f'install {target}', desc=f'{name} for PBRemTools')
if not launch.is_installed("segment_anything"):
launch.run_pip("install git+https://github.com/facebookresearch/segment-anything.git", desc="segment_anything for PBRemTools")