From 82add38ae150848bb982195480caf2bf1d232380 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Sat, 1 Jun 2024 21:00:47 +0100 Subject: [PATCH] TensorRT index fix --- build/COPY_ROOT/opt/ai-dock/bin/build/layer0/nvidia.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/nvidia.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/nvidia.sh index 666d799..b116ab2 100755 --- a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/nvidia.sh +++ b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/nvidia.sh @@ -17,7 +17,8 @@ build_nvidia_install_kohya_ss() { build_common_install_kohya_ss cd /opt/kohya_ss - micromamba run -n kohya_ss $PIP_INSTALL tensorflow tensorboard tensorrt==10.0.1 \ + micromamba run -n kohya_ss $PIP_INSTALL tensorrt==10.0.1 -extra-index-url https://pypi.nvidia.com + micromamba run -n kohya_ss $PIP_INSTALL tensorflow tensorboard \ -r requirements_linux_docker.txt ln -s /opt/micromamba/envs/kohya_ss/lib/python${PYTHON_VERSION}/site-packages/tensorrt_libs/libnvinfer.so.10 \