fix: remove the version limit of pynvml (#5068)

* fix: fix uv bin path for 0.5.0 Signed-off-by: Frost Ming <me@frostming.com> * fix: remove the version limit of pynvml Signed-off-by: Frost Ming <me@frostming.com> --------- Signed-off-by: Frost Ming <me@frostming.com>
bentoml · Nov 8, 2024 · e7d0ba2 · e7d0ba2
1 parent 0088c75
commit e7d0ba2
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 10 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,7 +37,7 @@ dependencies = [
     "httpx",
     "inflection",
     "numpy",
-    "nvidia-ml-py<12",
+    "nvidia-ml-py",
     # OpenTelemetry is the server dependencies, rather than SDK
     # Since there are discrepancies among API and instrumentation packages,
     # we should always pin the set version of Opentelemetry suite

diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_alpine.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_alpine.j2
@@ -10,7 +10,7 @@ ENV ENV /root/.bashrc
 
 ENV UV_SYSTEM_PYTHON=1
 RUN curl -LO https://astral.sh/uv/install.sh && \
-    sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/
+    sh install.sh && rm install.sh && mv $HOME/.local/bin/uv /usr/local/bin/
 
 {% if __options__system_packages is not none %}
 # Install user-defined system package

diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_amazonlinux.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_amazonlinux.j2
@@ -15,7 +15,7 @@ USER root
 
 ENV UV_SYSTEM_PYTHON=1
 RUN curl -LO https://astral.sh/uv/install.sh && \
-    sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/
+    sh install.sh && rm install.sh && mv $HOME/.local/bin/uv /usr/local/bin/
 {% if __options__system_packages is not none %}
 # Install user-defined system package
 {% call common.RUN(__enable_buildkit__) -%} {{ common.mount_cache("/var/cache/yum") }} {% endcall -%} set -eux && \

diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_debian.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_debian.j2
@@ -14,5 +14,5 @@ RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloa
 
 ENV UV_SYSTEM_PYTHON=1
 RUN curl -LO https://astral.sh/uv/install.sh && \
-    sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/
+    sh install.sh && rm install.sh && mv $HOME/.local/bin/uv /usr/local/bin/
 {% endblock %}
diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_ubi8.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_ubi8.j2
@@ -9,7 +9,7 @@ USER root
 
 ENV UV_SYSTEM_PYTHON=1
 RUN curl -LO https://astral.sh/uv/install.sh && \
-    sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/
+    sh install.sh && rm install.sh && mv $HOME/.local/bin/uv /usr/local/bin/
 {% if __options__system_packages is not none %}
 # Install user-defined system package
 {% call common.RUN(__enable_buildkit__) -%} {{ common.mount_cache("/var/cache/yum") }} {% endcall -%} set -eux && \

diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_v2.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_v2.j2
@@ -66,7 +66,7 @@ RUN {{ command }}
 {% endfor %}
 
 RUN curl -LO https://astral.sh/uv/install.sh && \
-    sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/ && uv venv --python {{ __options__python_version }}
+    sh install.sh && rm install.sh && mv $HOME/.local/bin/uv /usr/local/bin/ && uv venv --python {{ __options__python_version }}
 {% set __pip_cache__ = common.mount_cache("/root/.cache/") %}
 {% if __pip_preheat_packages__ %}
 {% for value in __pip_preheat_packages__ -%}