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

fix: numpy

Browse Source
This commit is contained in:
wtongze
2024-09-12 14:19:35 -07:00
parent c576633fe0
commit 4c55f8ec71
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+2
View File
@@ -6,6 +6,7 @@ ARG ROCM_VERSION="5.7.3"
ARG PYTORCH_BRANCH="v2.2.2" ARG PYTORCH_BRANCH="v2.2.2"
ARG TORCHVISION_BRANCH="v0.17.2" ARG TORCHVISION_BRANCH="v0.17.2"
ARG PYTHON_VERSION="3.11" ARG PYTHON_VERSION="3.11"
ARG NUMPY_VERSION="1.26.4"
ARG HIP_ARCH="gfx803;gfx900" ARG HIP_ARCH="gfx803;gfx900"
ARG DEBIAN_FRONTEND="noninteractive" ARG DEBIAN_FRONTEND="noninteractive"
@@ -29,6 +30,7 @@ EOF
RUN python${PYTHON_VERSION} -m pip install wheel RUN python${PYTHON_VERSION} -m pip install wheel
RUN python${PYTHON_VERSION} -m pip install -r requirements.txt RUN python${PYTHON_VERSION} -m pip install -r requirements.txt
RUN python${PYTHON_VERSION} -m pip install numpy==${NUMPY_VERSION}
RUN <<EOF RUN <<EOF
sudo apt-get update sudo apt-get update