mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build/Dockerfile

36 lines
1.5 KiB
Docker
Raw Normal View History

Update Docker build to produce manylinux2010 compliant wheels for non-cuda builds. Previously we lied claimed our wheels were manylinux1 compliant but they weren't. Uses a cross-compilation toolchain from the TF folks that builds manylinux2010 compliant wheels from a Ubuntu 16.04 VM. The CUDA wheels still aren't manylinux2010 compliant because they depend on CUDA libraries from the system.
2019-08-13 14:41:43 -04:00
FROM gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010
check out build files
2018-12-06 20:04:05 -08:00
LABEL maintainer "Matt Johnson <mattjj@google.com>"
Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.
2019-01-17 15:13:41 -05:00
WORKDIR /
Tweak Dockerfile to prevent build failure and add TODO (#3838)
2020-07-23 13:08:06 -07:00
# TODO(skyewm): delete the following line when no longer necessary.
RUN rm -f /etc/apt/sources.list.d/jonathonf-ubuntu-python-3_6-xenial.list
jaxlib build improvements (#1742)
2019-11-21 16:25:24 -08:00
RUN apt-get update
RUN apt-get install libffi-dev
jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support The TF/XLA build no longer works with CUDA 10.0. This could potentially be fixed, but isn't easy or officially supported.
2020-11-12 18:31:06 +00:00
RUN git clone --branch v1.2.21 https://github.com/pyenv/pyenv.git /pyenv
Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.
2019-01-17 15:13:41 -05:00
ENV PYENV_ROOT /pyenv
RUN /pyenv/bin/pyenv install 3.7.2
Update jaxlib build scripts to build Python 3.8.0 wheels. (#1612)
2019-10-31 11:46:37 -04:00
RUN /pyenv/bin/pyenv install 3.8.0
jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support The TF/XLA build no longer works with CUDA 10.0. This could potentially be fixed, but isn't easy or officially supported.
2020-11-12 18:31:06 +00:00
RUN /pyenv/bin/pyenv install 3.9.0
Update jaxlib build scripts to build Python 3.8.0 wheels. (#1612)
2019-10-31 11:46:37 -04:00
Update minimum NumPy version in jaxlib build scripts.
2021-08-02 09:22:32 -04:00
# We pin numpy to the minimum permitted version to avoid compatibility issues.
Drop support for NumPy 1.18. Per NEP-29, we can drop NumPy 1.18 support on Dec 22, 2021. The next NumPy deprecation will be 1.19 on Jun 21, 2022. PiperOrigin-RevId: 419651428
2022-01-04 12:11:00 -08:00
RUN eval "$(/pyenv/bin/pyenv init -)" && /pyenv/bin/pyenv local 3.7.2 && pip install numpy==1.19.5 setuptools wheel six auditwheel
RUN eval "$(/pyenv/bin/pyenv init -)" && /pyenv/bin/pyenv local 3.8.0 && pip install numpy==1.19.5 setuptools wheel six auditwheel
RUN eval "$(/pyenv/bin/pyenv init -)" && /pyenv/bin/pyenv local 3.9.0 && pip install numpy==1.19.5 setuptools wheel six auditwheel
Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.
2019-01-17 15:13:41 -05:00
Update Docker build to produce manylinux2010 compliant wheels for non-cuda builds. Previously we lied claimed our wheels were manylinux1 compliant but they weren't. Uses a cross-compilation toolchain from the TF folks that builds manylinux2010 compliant wheels from a Ubuntu 16.04 VM. The CUDA wheels still aren't manylinux2010 compliant because they depend on CUDA libraries from the system.
2019-08-13 14:41:43 -04:00
# Change the CUDA version if it doesn't match the installed version.
ARG JAX_CUDA_VERSION=10.0
Build CUDA 10.2 jaxlibs. (#2121) Also adds install_cuda.sh script that sets appropriate nccl and cuDNN versions.
2020-01-29 10:47:17 -08:00
COPY install_cuda.sh /install_cuda.sh
RUN chmod +x /install_cuda.sh
Update Docker build to produce manylinux2010 compliant wheels for non-cuda builds. Previously we lied claimed our wheels were manylinux1 compliant but they weren't. Uses a cross-compilation toolchain from the TF folks that builds manylinux2010 compliant wheels from a Ubuntu 16.04 VM. The CUDA wheels still aren't manylinux2010 compliant because they depend on CUDA libraries from the system.
2019-08-13 14:41:43 -04:00
RUN /bin/bash -c 'if [[ ! "$CUDA_VERSION" =~ ^$JAX_CUDA_VERSION.*$ ]]; then \
Build CUDA 10.2 jaxlibs. (#2121) Also adds install_cuda.sh script that sets appropriate nccl and cuDNN versions.
2020-01-29 10:47:17 -08:00
/install_cuda.sh $JAX_CUDA_VERSION; \
Update Docker build to produce manylinux2010 compliant wheels for non-cuda builds. Previously we lied claimed our wheels were manylinux1 compliant but they weren't. Uses a cross-compilation toolchain from the TF folks that builds manylinux2010 compliant wheels from a Ubuntu 16.04 VM. The CUDA wheels still aren't manylinux2010 compliant because they depend on CUDA libraries from the system.
2019-08-13 14:41:43 -04:00
fi'
Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.
2019-01-17 15:13:41 -05:00
check out build files
2018-12-06 20:04:05 -08:00
WORKDIR /
Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.
2019-01-17 15:13:41 -05:00
COPY build_wheel_docker_entrypoint.sh /build_wheel_docker_entrypoint.sh
check out build files
2018-12-06 20:04:05 -08:00
RUN chmod +x /build_wheel_docker_entrypoint.sh
WORKDIR /build
ENV TEST_TMPDIR /build
ENTRYPOINT ["/build_wheel_docker_entrypoint.sh"]
Reference in New Issue Copy Permalink