mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #11780 from ROCmSoftwarePlatform:rocm_update_dockerfile

Browse Source 
PiperOrigin-RevId: 466756858
This commit is contained in:
jax authors 2022-08-10 12:19:13 -07:00
commit e81578a9fa
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
build/rocm/Dockerfile.rocm
View File

@ -85,7 +85,6 @@ RUN add-apt-repository ppa:deadsnakes/ppa && \
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 && \
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 2
RUN pip3 install --upgrade setuptools pip
RUN pip3 install absl-py numpy==1.19.5 scipy wheel six setuptools pytest pytest-rerunfailures
RUN pip3 install --upgrade --force-reinstall setuptools pip
RUN pip3 install absl-py numpy==1.20.0 scipy wheel six setuptools pytest pytest-rerunfailures matplotlib