File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -45,9 +45,9 @@ gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key" > /etc/yum.repos.d/amdgpu.repo
45
45
RUN --mount=type=cache,target=/root/.cache/uv \
46
46
export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
47
47
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} && \
51
51
# Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
52
52
microdnf install -y --nodocs libdrm-amdgpu && \
53
53
microdnf clean all
You can’t perform that action at this time.
0 commit comments