wtongze/rocm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: python version

Browse Source
This commit is contained in:
wtongze
2024-09-10 15:26:27 -07:00
parent 1476c39588
commit d1a82edbaf
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+10 -3
View File
@@ -17,14 +17,21 @@ RUN git submodule update --init --recursive
RUN <<EOF
sudo apt-get update
sudo apt install -y python3-dev
sudo apt install -y software-properties-common
sudo add-apt-repository -y ppa:deadsnakes/ppa
sudo apt-get update
sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev
sudo apt-get clean
sudo rm -rf /var/lib/apt/lists/*
EOF
RUN pip install typing-extensions
RUN python${PYTHON_VERSION} -m pip install typing-extensions wheel
ENV PYTORCH_ROCM_ARCH=${HIP_ARCH}
RUN python3 tools/amd_build/build_amd.py && USE_ROCM=1 python3 setup.py bdist_wheel
RUN python${PYTHON_VERSION} tools/amd_build/build_amd.py && USE_ROCM=1 python${PYTHON_VERSION} setup.py bdist_wheel
WORKDIR /home/rocm/pytorch/dist
RUN cp (ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/pytorch-rocm.whl
WORKDIR /home/rocm
CMD [ "bash", "-l" ]