forked from mirrors/nixpkgs
a730888ce0
Major breaking change from 1.x, so treating keeping both versions for now. (adapted from33f11be707) (adapted from9e8dea7986)
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
|