Skip to content

Commit

Permalink
Use pip install vs wgets/rm -rf *.whl
Browse files Browse the repository at this point in the history
  • Loading branch information
Ted Themistokleous committed Feb 8, 2024
1 parent 3ca051a commit 47836f1
Showing 1 changed file with 4 additions and 9 deletions.
13 changes: 4 additions & 9 deletions tools/docker/migraphx_with_onnxruntime_pytorch.docker
Original file line number Diff line number Diff line change
Expand Up @@ -8,22 +8,17 @@ RUN apt update && apt install -y wget
#Aquire and install ROCm
RUN wget https://repo.radeon.com/amdgpu-install/6.0.2/ubuntu/jammy/amdgpu-install_6.0.60002-1_all.deb
RUN apt install -y ./amdgpu-install_6.0.60002-1_all.deb
RUN amdgpu-install --usecase=rocm -y && rm *.deb
RUN amdgpu-install --usecase=rocm -y && rm amdgpu-install_6.0.60002-1_all.deb

#Install MIGraphX from package manager
RUN apt install -y migraphx

#Pieces for Onnxruntime for ROCm and MIGraphX Execution Provider Support
RUN wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0.2/onnxruntime_rocm-inference-1.17.0-cp310-cp310-linux_x86_64.whl
RUN pip3 install *.whl && rm -rf *.whl
RUN pip3 install https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0.2/onnxruntime_rocm-inference-1.17.0-cp310-cp310-linux_x86_64.whl

#Pieces for pytorch
RUN wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0/torch-2.1.1+rocm6.0-cp310-cp310-linux_x86_64.whl
RUN pip3 install *.whl && rm -rf *.whl

RUN wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0/torchvision-0.16.1+rocm6.0-cp310-cp310-linux_x86_64.whl
RUN pip3 install *.whl && rm -rf *.whl

RUN pip3 install https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0/torch-2.1.1+rocm6.0-cp310-cp310-linux_x86_64.whl
RUN pip3 install https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0/torchvision-0.16.1+rocm6.0-cp310-cp310-linux_x86_64.whl

#Adjust final path for ability to use rocm components
ENV PATH=$PATH:/opt/rocm/bin/
Expand Down

0 comments on commit 47836f1

Please sign in to comment.