This repo allows to produce benchmarks for different Machine Learning models created for different tasks written in different languages (Keras, Pytorch, Swift).
To setup full working environment you need:
- Swift for Tensorflow toolchain
- Anaconda setup for the installation of proper packages
# install anconda
curl -ksLO "https://repo.anaconda.com/archive/Anaconda3-2020.07-MacOSX-x86_64.sh"
sh Anaconda3-2020.07-MacOSX-x86_64.sh -u
# setup proper environment and install python dependencies
conda create -n swift-env tensorflow
conda activate swift-env
conda install -c conda-forge matplotlib
conda install pytorch torchvision torchaudio -c pytorch
# setup proper PATHs environments for swift usage of python
# please change /opt/anaconda3 to whatever path you may have
export PATH=/opt/anaconda3/bin:$PATH
export PATH=/Library/Developer/Toolchains/swift-latest/usr/bin:$PATH
export PYTHON_LIBRARY=/opt/anaconda3/envs/uproot/lib/libpython3.8.dylib
export PYTHONPATH=/opt/anaconda3/envs/uproot/lib/python3.8:/opt/anaconda3/envs/uproot/lib/python3.8/lib-dynload:/opt/anaconda3/envs/uproot/lib/python3.8/site-packages
Then you clone the repo and start using it.
git clone https://github.com/lgiommi/Swift4TFBenchmarks