tjh.dev
1#!/usr/bin/env bash
2
3version=1.14.0
4hashfile=binary-hashes.nix
5rm -f $hashfile
6echo "{" >> $hashfile
7echo "version = \"$version\";" >> $hashfile
8for sys in "linux" "mac"; do
9 for tfpref in "cpu/tensorflow" "gpu/tensorflow_gpu"; do
10 for pykind in "py2-none-any" "py3-none-any" "cp27-none-linux_x86_64" "cp35-cp35m-linux_x86_64" "cp36-cp36m-linux_x86_64" "cp37-cp37m-linux_x86_64"; do
11 if [ $sys == "mac" ]; then
12 [[ $pykind =~ py.* ]] && [[ $tfpref =~ cpu.* ]]
13 result=$?
14 pyver=${pykind:2:1}
15 flavour=cpu
16 else
17 [[ $pykind =~ .*linux.* ]]
18 result=$?
19 pyver=${pykind:2:2}
20 flavour=${tfpref:0:3}
21 fi
22 if [ $result == 0 ]; then
23 url=https://storage.googleapis.com/tensorflow/$sys/$tfpref-$version-$pykind.whl
24 hash=$(nix-prefetch-url $url)
25 echo "${sys}_py_${pyver}_${flavour} = {" >> $hashfile
26 echo " url = \"$url\";" >> $hashfile
27 echo " sha256 = \"$hash\";" >> $hashfile
28 echo "};" >> $hashfile
29 fi
30 done
31 done
32done
33echo "}" >> $hashfile