holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use nvidia pypi index url for linux

Browse Source
This commit is contained in:
cmdr2 2023-07-29 22:24:34 +05:30
parent 14679586a8
commit 1e2c9ecb41
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ui/easydiffusion/package_manager.py
View File

@ -11,7 +11,11 @@ from easydiffusion import app
manifest = { manifest = {
"tensorrt": { "tensorrt": {
"install": ["nvidia-cudnn", "tensorrt-libs", "tensorrt"], "install": [
"nvidia-cudnn --extra-index-url=https://pypi.ngc.nvidia.com --trusted-host pypi.ngc.nvidia.com",
"tensorrt-libs --extra-index-url=https://pypi.ngc.nvidia.com --trusted-host pypi.ngc.nvidia.com",
"tensorrt --extra-index-url=https://pypi.ngc.nvidia.com --trusted-host pypi.ngc.nvidia.com",
],
"uninstall": ["tensorrt"], "uninstall": ["tensorrt"],
# TODO also uninstall tensorrt-libs and nvidia-cudnn, but do it upon restarting (avoid 'file in use' error) # TODO also uninstall tensorrt-libs and nvidia-cudnn, but do it upon restarting (avoid 'file in use' error)
} }
@ -29,6 +33,7 @@ if platform.system() == "Windows":
wheels = [] wheels = []
for p in packages: for p in packages:
p = p.split(" ")[0]
f = next((f for f in files if f.startswith(p) and f.endswith((".whl", ".tar.gz"))), None) f = next((f for f in files if f.startswith(p) and f.endswith((".whl", ".tar.gz"))), None)
if f: if f:
wheels.append(os.path.join(trt_dir, f)) wheels.append(os.path.join(trt_dir, f))