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

jaxlib build fixes. (#4066)

Browse Source 
1. `wheel.pep425tags` has been removed as of
   https://github.com/pypa/setuptools/pull/1829. Use the new
   `packaging.tags` instead.

2. Add `--allow-downgrades` to cuda install command. I'm not sure this
   is always necessary, but I ran into it, I'm guessing due to a cached
   docker image.
This commit is contained in:
Skye Wanderman-Milne 2020-09-08 18:23:42 -07:00 committed by GitHub
parent 4600dd7957
commit cf2d15d4bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
build/build_wheel_docker_entrypoint.sh
View File

@ -34,7 +34,7 @@ fi
# Builds and activates a specific Python version.
pyenv local "$PY_VERSION"
PY_TAG=$(python -c "import wheel; import wheel.pep425tags as t; print(t.get_abbr_impl() + t.get_impl_ver())")
PY_TAG=$(python -c "import packaging.tags as t; print(t.interpreter_name() + t.interpreter_version())")
echo "Python tag: $PY_TAG"

2
build/install_cuda.sh
View File

@ -19,7 +19,7 @@ echo "cudnn version: $CUDNN_VERSION"
apt-get update
apt-get remove -y --allow-change-held-packages cuda-license-10-0 libcudnn7 libnccl2
apt-get install -y --no-install-recommends \
apt-get install -y --no-install-recommends --allow-downgrades \
cuda-nvml-dev-$CUDA_VERSION \
cuda-command-line-tools-$CUDA_VERSION \
cuda-libraries-dev-$CUDA_VERSION \