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

fix: path

Browse Source
This commit is contained in:
wtongze
2024-09-10 14:05:42 -07:00
parent 9dc3b7ac97
commit 227c721385
2 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base.dockerfile
+1 -1
View File
@@ -86,7 +86,7 @@ EOF
# Change to non-root user # Change to non-root user
USER rocm USER rocm
WORKDIR /home/rocm WORKDIR /home/rocm
ENV PATH="${PATH}:/opt/rocm/bin" ENV PATH="${PATH}:/opt/rocm/bin:/home/rocm/.local/bin"
# Default to login shell # Default to login shell
CMD [ "bash", "-l" ] CMD [ "bash", "-l" ]
tensorflow.dockerfile
-2
View File
@@ -32,8 +32,6 @@ sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev
sudo rm -rf /var/lib/apt/lists/* sudo rm -rf /var/lib/apt/lists/*
EOF EOF
ENV PATH=$PATH:/home/rocm/.local/bin
# TODO: simplify dependencies # TODO: simplify dependencies
# > Ref: https://github.com/ROCm/ROCm/issues/1880 # > Ref: https://github.com/ROCm/ROCm/issues/1880
RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf