a730888ce0
Major breaking change from 1.x, so treating keeping both versions for now. (adapted from 33f11be707e39acf96423f97f3baa80d8f11a0cb) (adapted from 9e8dea7986dbdde850a58c7704182776642d8919)
34 lines
1.2 KiB
Bash
Executable File
34 lines
1.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
version=1.14.0
|
|
hashfile=binary-hashes.nix
|
|
rm -f $hashfile
|
|
echo "{" >> $hashfile
|
|
echo "version = \"$version\";" >> $hashfile
|
|
for sys in "linux" "mac"; do
|
|
for tfpref in "cpu/tensorflow" "gpu/tensorflow_gpu"; do
|
|
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
|
|
if [ $sys == "mac" ]; then
|
|
[[ $pykind =~ py.* ]] && [[ $tfpref =~ cpu.* ]]
|
|
result=$?
|
|
pyver=${pykind:2:1}
|
|
flavour=cpu
|
|
else
|
|
[[ $pykind =~ .*linux.* ]]
|
|
result=$?
|
|
pyver=${pykind:2:2}
|
|
flavour=${tfpref:0:3}
|
|
fi
|
|
if [ $result == 0 ]; then
|
|
url=https://storage.googleapis.com/tensorflow/$sys/$tfpref-$version-$pykind.whl
|
|
hash=$(nix-prefetch-url $url)
|
|
echo "${sys}_py_${pyver}_${flavour} = {" >> $hashfile
|
|
echo " url = \"$url\";" >> $hashfile
|
|
echo " sha256 = \"$hash\";" >> $hashfile
|
|
echo "};" >> $hashfile
|
|
fi
|
|
done
|
|
done
|
|
done
|
|
echo "}" >> $hashfile
|