Svyatoslav Scherbina
20 lines
698 B
Bash
Executable File
20 lines
698 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
KONAN_USER_DIR=${KONAN_DATA_DIR:-"$HOME/.konan"}
|
|
TF_TARGET_DIRECTORY="$KONAN_USER_DIR/third-party/tensorflow"
|
|
TF_TYPE="cpu" # Change to "gpu" for GPU support
|
|
|
|
if [ x$TARGET == x ]; then
|
|
case "$OSTYPE" in
|
|
darwin*) TARGET=macbook; TF_TARGET=darwin ;;
|
|
linux*) TARGET=linux; TF_TARGET=linux ;;
|
|
*) echo "unknown: $OSTYPE" && exit 1;;
|
|
esac
|
|
fi
|
|
|
|
if [ ! -d $TF_TARGET_DIRECTORY/include/tensorflow ]; then
|
|
echo "Installing TensorFlow into $TF_TARGET_DIRECTORY ..."
|
|
mkdir -p $TF_TARGET_DIRECTORY
|
|
curl -s -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-${TF_TARGET}-x86_64-1.1.0.tar.gz" | tar -C $TF_TARGET_DIRECTORY -xz
|
|
fi
|