feat: build files

pytorch.dockerfile

@@ -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/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4
 
-# Test
 RUN <<EOF
 sudo apt-get update
 sudo apt-get install -y ffmpeg
@@ -73,7 +72,13 @@ WORKDIR /home/rocm/audio
 RUN USE_ROCM=ON \
 PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") \
 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
 CMD [ "bash", "-l" ]