mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-17 15:02:57 +00:00
b2efa3cd8c
* Update JULIA_CPU_TARGET for precompiling packages * Update links * Also update link for aarch64
56 lines
2.4 KiB
Bash
Executable File
56 lines
2.4 KiB
Bash
Executable File
#!/bin/bash
|
|
set -exuo pipefail
|
|
# Requirements:
|
|
# - Run as a non-root user
|
|
# - The JULIA_PKGDIR environment variable is set
|
|
# - Julia is already set up, with the setup_julia.py command
|
|
|
|
|
|
# If we don't specify what CPUs the precompilation should be done for, it's
|
|
# *only* done for the target of the host doing the compilation. When the
|
|
# container runs on a host that's the same architecture, but a *different*
|
|
# generation of CPU than what the build host was, the precompilation is useless
|
|
# and Julia takes a long long time to start up. This specific multitarget comes
|
|
# from https://github.com/JuliaCI/julia-buildkite/blob/9f354745a1f2bf31a5952462aa1ff2d869507cb8/utilities/build_envs.sh#L20-L82,
|
|
# and may need to be updated as new CPU generations come out.
|
|
# If the architecture the container runs on is different,
|
|
# precompilation may still have to be re-done on first startup - but this
|
|
# *should* catch most of the issues. See
|
|
# https://github.com/jupyter/docker-stacks/issues/2015 for more information
|
|
if [ "$(uname -m)" == "x86_64" ]; then
|
|
# See https://github.com/JuliaCI/julia-buildkite/blob/9f354745a1f2bf31a5952462aa1ff2d869507cb8/utilities/build_envs.sh#L23
|
|
# for an explanation of these options
|
|
export JULIA_CPU_TARGET="generic;sandybridge,-xsaveopt,clone_all;haswell,-rdrnd,base(1);x86-64-v4,-rdrnd,base(1)"
|
|
elif [ "$(uname -m)" == "aarch64" ]; then
|
|
# See https://github.com/JuliaCI/julia-buildkite/blob/9f354745a1f2bf31a5952462aa1ff2d869507cb8/utilities/build_envs.sh#L56
|
|
# for an explanation of these options
|
|
export JULIA_CPU_TARGET="generic;cortex-a57;thunderx2t99;carmel,clone_all;apple-m1,base(3);neoverse-512tvb,base(3)"
|
|
fi
|
|
|
|
# Install base Julia packages
|
|
julia -e '
|
|
import Pkg;
|
|
Pkg.update();
|
|
Pkg.add([
|
|
"HDF5",
|
|
"IJulia",
|
|
"Pluto"
|
|
]);
|
|
Pkg.precompile();
|
|
'
|
|
|
|
# Move the kernelspec out of ${HOME} to the system share location.
|
|
# Avoids problems with runtime UID change not taking effect properly
|
|
# on the .local folder in the jovyan home dir.
|
|
mv "${HOME}/.local/share/jupyter/kernels/julia"* "${CONDA_DIR}/share/jupyter/kernels/"
|
|
chmod -R go+rx "${CONDA_DIR}/share/jupyter"
|
|
rm -rf "${HOME}/.local"
|
|
fix-permissions "${JULIA_PKGDIR}" "${CONDA_DIR}/share/jupyter"
|
|
|
|
# Install jupyter-pluto-proxy to get Pluto to work on JupyterHub
|
|
mamba install --yes \
|
|
'jupyter-pluto-proxy' && \
|
|
mamba clean --all -f -y && \
|
|
fix-permissions "${CONDA_DIR}" && \
|
|
fix-permissions "/home/${NB_USER}"
|