Commit 059696a8 authored by hacker1024's avatar hacker1024
Browse files

python310Packages.tensorflow-bin: Add Jetson source

parent c3aa7b89

pkgs/development/python-modules/tensorflow/binary-hashes.nix

+5 −0
Original line number Diff line number Diff line 
{

  version = "2.17.0";

  version_jetson = "2.16.1+nv24.07";

  x86_64-linux_39 = {

    url = "https://storage.googleapis.com/tensorflow/versions/2.17.0/tensorflow_cpu-2.17.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";

    sha256 = "1aacn68b88bnnmpl1q0irih0avzm2lfyhwr3wldg144n5zljlrbx";
@@ -48,6 +49,10 @@ 
    url = "https://storage.googleapis.com/tensorflow/versions/2.17.0/tensorflow-2.17.0-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl";

    sha256 = "1zrscms9qkfpiscnl8c7ibfipwpw8jrdfvwh4wb69p9rxvqgxbbj";

  };

  aarch64-linux_310_jetson = {

    url = "https://developer.download.nvidia.com/compute/redist/jp/v60/tensorflow/tensorflow-2.16.1+nv24.07-cp310-cp310-linux_aarch64.whl";

    sha256 = "1ymdknl5v41z6z0wg068diici30am8vysg6b6sqxr8w6yk4aib42";

  };

  aarch64-darwin_39 = {

    url = "https://storage.googleapis.com/tensorflow/versions/2.17.0/tensorflow-2.17.0-cp39-cp39-macosx_12_0_arm64.whl";

    sha256 = "01a3hjnrgjp2i0ciwyy0gki41cy32prvjhr20zhlcjwbssarxy4p";

pkgs/development/python-modules/tensorflow/prefetcher.sh

+4 −0
Original line number Diff line number Diff line 
#!/usr/bin/env bash



version="2.17.0"

version_jetson="2.16.1+nv24.07"



bucket="https://storage.googleapis.com/tensorflow/versions/${version}"

bucket_jetson="https://developer.download.nvidia.com/compute/redist/jp/v60/tensorflow"



# List of binary wheels for Tensorflow.  The most recent versions can be found

# on the following page:
@@ -20,6 +22,7 @@ url_and_key_list=( 
"aarch64-linux_310 $bucket/tensorflow-${version}-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl"

"aarch64-linux_311 $bucket/tensorflow-${version}-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl"

"aarch64-linux_312 $bucket/tensorflow-${version}-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl"

"aarch64-linux_310_jetson $bucket_jetson/tensorflow-${version_jetson}-cp310-cp310-linux_aarch64.whl"

"aarch64-darwin_39 $bucket/tensorflow-${version}-cp39-cp39-macosx_12_0_arm64.whl"

"aarch64-darwin_310 $bucket/tensorflow-${version}-cp310-cp310-macosx_12_0_arm64.whl"

"aarch64-darwin_311 $bucket/tensorflow-${version}-cp311-cp311-macosx_12_0_arm64.whl"
@@ -30,6 +33,7 @@ hashfile=binary-hashes.nix 
rm -f $hashfile

echo "{" >> $hashfile

echo "version = \"$version\";" >> $hashfile

echo "version_jetson = \"$version_jetson\";" >> $hashfile



for url_and_key in "${url_and_key_list[@]}"; do

  key=$(echo "$url_and_key" | cut -d' ' -f1)