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

fix: torch wheel

Browse Source
This commit is contained in:
wtongze
2024-09-12 13:09:26 -07:00
parent 6c902d6f75
commit 6bd54aa318
1 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+1 -10
View File
@@ -48,9 +48,6 @@ PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") python${PYTHON_VER
# https://github.com/pytorch/pytorch/issues/125230 # https://github.com/pytorch/pytorch/issues/125230
# https://github.com/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4 # https://github.com/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4
WORKDIR /home/rocm/pytorch/dist
RUN cp $(ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/torch.whl
# Test # Test
RUN <<EOF RUN <<EOF
sudo apt-get update sudo apt-get update
@@ -59,7 +56,7 @@ sudo apt-get clean
sudo rm -rf /var/lib/apt/lists/* sudo rm -rf /var/lib/apt/lists/*
EOF EOF
RUN python${PYTHON_VERSION} -m pip install /home/rocm/torch.whl RUN python${PYTHON_VERSION} -m pip install /home/rocm/pytorch/dist/*.whl
# Torch Vision # Torch Vision
WORKDIR /home/rocm/ WORKDIR /home/rocm/
@@ -68,9 +65,6 @@ RUN git clone --depth 1 https://github.com/pytorch/vision.git -b v0.17.2 vision
WORKDIR /home/rocm/vision WORKDIR /home/rocm/vision
RUN USE_ROCM=ON python${PYTHON_VERSION} setup.py bdist_wheel RUN USE_ROCM=ON python${PYTHON_VERSION} setup.py bdist_wheel
WORKDIR /home/rocm/vision/dist
RUN cp $(ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/torchvision.whl
# Torch Audio # Torch Audio
WORKDIR /home/rocm/ WORKDIR /home/rocm/
RUN git clone --depth 1 https://github.com/pytorch/audio.git -b ${PYTORCH_BRANCH} audio RUN git clone --depth 1 https://github.com/pytorch/audio.git -b ${PYTORCH_BRANCH} audio
@@ -79,9 +73,6 @@ WORKDIR /home/rocm/audio
RUN USE_ROCM=ON \ RUN USE_ROCM=ON \
PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") \ PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") \
python${PYTHON_VERSION} setup.py bdist_wheel python${PYTHON_VERSION} setup.py bdist_wheel
WORKDIR /home/rocm/audio/dist
RUN cp $(ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/torchaudio.whl
# Test End # Test End
WORKDIR /home/rocm WORKDIR /home/rocm