diff --git a/requirements-importer-macos.txt b/requirements-importer-macos.txt new file mode 100644 index 00000000..a1e74389 --- /dev/null +++ b/requirements-importer-macos.txt @@ -0,0 +1,38 @@ +-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html +--pre + +numpy +torch +torchvision + +#iree-compiler | iree-runtime should already be installed +#these dont work ok osx +#iree-tools-tflite +#iree-tools-xla +#iree-tools-tf + +# TensorFlow and JAX. +gin-config +tensorflow-macos +tensorflow-metal +tf-models-nightly +tensorflow-text-nightly +transformers +jax[cpu] + +# tflitehub dependencies. +Pillow + +# Testing and support. +lit +pyyaml + +#ONNX and ORT for benchmarking +--extra-index-url https://test.pypi.org/simple/ +#protobuf +#coloredlogs +#flatbuffers +#sympy +#psutil +#onnx-weekly +#ort-nightly diff --git a/requirements-importer.txt b/requirements-importer.txt index f00e9a88..d152c1db 100644 --- a/requirements-importer.txt +++ b/requirements-importer.txt @@ -1,5 +1,11 @@ -#iree-compiler should already be installed -#iree-runtime +-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html +--pre + +numpy +torch +torchvision + +#iree-compiler | iree-runtime should already be installed iree-tools-tflite iree-tools-xla iree-tools-tf diff --git a/requirements.txt b/requirements.txt index 00560ca4..27f3a7b2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,11 +1,3 @@ --f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html ---pre - -numpy -torch -torchvision - -# Build requirements. setuptools wheel pytest diff --git a/setup_venv.sh b/setup_venv.sh index 1615c44d..2124cfda 100755 --- a/setup_venv.sh +++ b/setup_venv.sh @@ -100,8 +100,8 @@ if [[ ! -z "${IMPORTER}" ]]; then echo "${Yellow}Linux detected.. installing importer tools" $PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://test.pypi.org/simple/ elif [[ $(uname -s) = 'Darwin' ]]; then - #Conda seems to have some problems install these packages and hope they get resolved upstream. - $PYTHON -m pip install tensorflow-macos tensorflow-metal + #Conda seems to have some problems installing these packages and hope they get resolved upstream. + $PYTHON -m pip install --upgrade -r "$TD/requirements-importer-macos.txt" -f https://github.com/${RUNTIME}/releases fi fi