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

chore: type

Browse Source
This commit is contained in:
wtongze
2024-09-12 13:49:13 -07:00
parent 4fc0937174
commit c576633fe0
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
@@ -43,7 +43,7 @@ ENV PYTORCH_BUILD_NUMBER=0
# Torch # Torch
# > 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
RUN python${PYTHON_VERSION} tools/amd_build/build_amd.py && \ RUN python${PYTHON_VERSION} tools/amd_build/build_amd.py && \
USE_ROCM=ON \ USE_ROCM=ON \
USE_NUMPY=1 \ USE_NUMPY=1 \