Skip to content

Commit b357f89

Browse files
authored
Dockerfile.rocm.ubi: use default torch index ROCm (do not use nightlies anymore) (opendatahub-io#220)
1 parent 88c65ab commit b357f89

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Dockerfile.rocm.ubi

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,9 @@ gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key" > /etc/yum.repos.d/amdgpu.repo
4545
RUN --mount=type=cache,target=/root/.cache/uv \
4646
export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
4747
uv pip install --pre \
48-
--index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
49-
torch==2.7.0.dev20250308+rocm${version}\
50-
torchvision==0.22.0.dev20250308+rocm${version} && \
48+
--index-url "https://download.pytorch.org/whl/ \
49+
torch==2.7.0+rocm${version}\
50+
torchvision==0.22.0+rocm${version} && \
5151
# Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
5252
microdnf install -y --nodocs libdrm-amdgpu && \
5353
microdnf clean all

0 commit comments

Comments
 (0)