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

feat: build files

Browse Source
This commit is contained in:
wtongze
2024-09-12 13:13:01 -07:00
parent 6bd54aa318
commit a69380acdd
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+7 -2
View File
@@ -48,7 +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
# Test
RUN <<EOF RUN <<EOF
sudo apt-get update sudo apt-get update
sudo apt-get install -y ffmpeg sudo apt-get install -y ffmpeg
@@ -73,7 +72,13 @@ 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
# Test End
RUN <<EOF
mkdir /home/rocm/build
cp /home/rocm/pytorch/dist/*.whl /home/rocm/build
cp /home/rocm/vision/dist/*.whl /home/rocm/build
cp /home/rocm/audio/dist/*.whl /home/rocm/build
EOF
WORKDIR /home/rocm WORKDIR /home/rocm
CMD [ "bash", "-l" ] CMD [ "bash", "-l" ]