mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-09 11:02:57 +00:00
b6104850fa
* Automatically install latest julia version * Better text * Fix * Fix * Update setup-julia.bash * Install plumbum * Better docs * Better docs * Use subprocess.check_call instead of plumbum * Do not use dash in python filename * Remove plumbum from the image * Remove jq from the image * Remove setup-julia.bash file * Fix file name * Fix docstring
86 lines
2.6 KiB
Python
Executable File
86 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
# Requirements:
|
|
# - Run as the root user
|
|
# - The JULIA_PKGDIR environment variable is set
|
|
|
|
import os
|
|
import platform
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
import requests
|
|
|
|
|
|
def unify_aarch64(platform: str) -> str:
|
|
"""
|
|
Renames arm64->aarch64 to support local builds on on aarch64 Macs
|
|
"""
|
|
return {
|
|
"aarch64": "aarch64",
|
|
"arm64": "aarch64",
|
|
"x86_64": "x86_64",
|
|
}[platform]
|
|
|
|
|
|
def get_latest_julia_url() -> tuple[str, str]:
|
|
"""
|
|
Get the last stable version of Julia
|
|
Based on: https://github.com/JuliaLang/www.julialang.org/issues/878#issuecomment-749234813
|
|
"""
|
|
|
|
versions = requests.get(
|
|
"https://julialang-s3.julialang.org/bin/versions.json"
|
|
).json()
|
|
stable_versions = {k: v for k, v in versions.items() if v["stable"]}
|
|
latest_version_files = stable_versions[max(stable_versions)]["files"]
|
|
triplet = unify_aarch64(platform.machine()) + "-linux-gnu"
|
|
file_info = [vf for vf in latest_version_files if vf["triplet"] == triplet][0]
|
|
return file_info["url"], file_info["version"]
|
|
|
|
|
|
def download_julia(julia_url: str) -> None:
|
|
"""
|
|
Downloads and unpacks julia
|
|
The resulting julia directory is "/opt/julia-VERSION/"
|
|
"""
|
|
tmp_file = Path("/tmp/julia.tar.gz")
|
|
subprocess.check_call(
|
|
["curl", "--progress-bar", "--location", "--output", tmp_file, julia_url]
|
|
)
|
|
shutil.unpack_archive(tmp_file, "/opt/")
|
|
tmp_file.unlink()
|
|
|
|
|
|
def prepare_julia(julia_version: str) -> None:
|
|
"""
|
|
Creates /usr/local/bin/julia symlink
|
|
Make Julia aware of conda libraries
|
|
Creates a directory for Julia user libraries
|
|
"""
|
|
# Link Julia installed version to /usr/local/bin, so julia launches it
|
|
subprocess.check_call(
|
|
["ln", "-fs", f"/opt/julia-{julia_version}/bin/julia", "/usr/local/bin/julia"]
|
|
)
|
|
|
|
# Tell Julia where conda libraries are
|
|
Path("/etc/julia").mkdir()
|
|
Path("/etc/julia/juliarc.jl").write_text(
|
|
f'push!(Libdl.DL_LOAD_PATH, "{os.environ["CONDA_DIR"]}/lib")\n'
|
|
)
|
|
|
|
# Create JULIA_PKGDIR, where user libraries are installed
|
|
JULIA_PKGDIR = Path(os.environ["JULIA_PKGDIR"])
|
|
JULIA_PKGDIR.mkdir()
|
|
subprocess.check_call(["chown", os.environ["NB_USER"], JULIA_PKGDIR])
|
|
subprocess.check_call(["fix-permissions", JULIA_PKGDIR])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
julia_url, julia_version = get_latest_julia_url()
|
|
download_julia(julia_url=julia_url)
|
|
prepare_julia(julia_version=julia_version)
|