git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-03-06 15:09:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

bugfix(uv) Lock torchvision and ensure the docker uses the same rocm version

Browse Source
This commit is contained in:
Heathen711 2025-07-05 03:35:11 +00:00
parent a3cb3e03f4
commit 0db304f1ee
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile
View File

@ -82,7 +82,7 @@ RUN --mount=type=cache,target=/var/cache/apt \
--mount=type=cache,target=/var/lib/apt \
if [ "$GPU_DRIVER" = "rocm" ]; then \
wget -O /tmp/amdgpu-install.deb \
https://repo.radeon.com/amdgpu-install/6.2.4/ubuntu/noble/amdgpu-install_6.2.60204-1_all.deb && \
https://repo.radeon.com/amdgpu-install/6.3.4/ubuntu/noble/amdgpu-install_6.3.60304-1_all.deb && \
apt install -y /tmp/amdgpu-install.deb && \
apt update && \
amdgpu-install --usecase=rocm -y && \

2
docker/run.sh
View File

@ -25,7 +25,7 @@ run() {
printf "%s\n" "$build_args"
fi
docker compose build $build_args $service_name
docker compose build --no-cache $build_args $service_name
unset build_args
printf "%s\n" "starting service $service_name"