aboutsummaryrefslogtreecommitdiff
path: root/nixpkgs/pkgs/development/libraries/science/math/tensorflow/prefetcher.sh
blob: 515f25df4c1f3696e520e9ceb7023d84c3d2fe49 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/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" "darwin"; do
    for tfpref in "cpu" "gpu"; do
        for platform in "x86_64"; do
            if [ $sys = "darwin" ] && [ $tfpref = "gpu" ]; then
               continue
            fi
            url=https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-$tfpref-$sys-$platform-$version.tar.gz
            hash=$(nix-prefetch-url $url)
            echo "\"${tfpref}-${sys}-${platform}\" = {" >> $hashfile
            echo "  url = \"$url\";" >> $hashfile
            echo "  sha256 = \"$hash\";" >> $hashfile
            echo "};" >> $hashfile
        done
    done
done
echo "}" >> $hashfile