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

feat: add numpy for pytorch

Browse Source
This commit is contained in:
wtongze
2024-09-11 18:32:29 -07:00
parent 535782308f
commit 7bd4c10b51
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+1 -1
View File
@@ -49,7 +49,7 @@ ENV PYTORCH_ROCM_ARCH=${HIP_ARCH}
ENV PYTORCH_BUILD_VERSION=${PYTORCH_BUILD_VERSION}
ENV PYTORCH_BUILD_NUMBER=0
RUN python${PYTHON_VERSION} tools/amd_build/build_amd.py && \
USE_ROCM=ON USE_FLASH_ATTENTION=OFF USE_MEM_EFF_ATTENTION=OFF python${PYTHON_VERSION} setup.py bdist_wheel
USE_ROCM=ON USE_NUMPY=1 USE_FLASH_ATTENTION=OFF USE_MEM_EFF_ATTENTION=OFF python${PYTHON_VERSION} setup.py bdist_wheel
# https://github.com/pytorch/pytorch/issues/125230
# https://github.com/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4