jetson-tensorflow-pytorch-build

Provides an environment for compiling TensorFlow or PyTorch with CUDA for aarch64 on an x86 machine. This is for Jetson. If you build using an EC2 m6g.16xlarge (aarch64) instance, TensorFlow can be fully built in about 30 minutes. It can be used as a cross-compilation environment not only for TensorFlow and PyTorch, but also for various other packages and libraries.

MIT License

Stars

13

View Code on GitHub Visit Website View on X

Ecosystems: PyTorch, Tensorflow, Docker, Cuda

Issue Statistics

Past Year

All Time

Total Pull Requests

0

0

Merged Pull Requests

0

0

Total Issues

0

0

Time to Close Issues

N/A

N/A

Related Projects

cxbqn

BQN virtual machine

nvidia-gpu-ml-library-test

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being p...

PyTorchTOP

GPU PyTorch TOP in TouchDesigner with CUDA-enabled OpenCV

dockerdl

Deep Learning Docker Image

docker_image_with_cuda10_cudnn7

Dockerfiles and manual for easy build of docker image with CUDA10.X and cuDNN7.6 to run TensorFlo...

docker-devel-env

Fast, reproducible, and portable software development environments

computeWorks_examples

Matrix multiplication example performed with OpenMP, OpenACC, BLAS, cuBLABS, and CUDA

tensorflow-cmake

TensorFlow examples in C, C++, Go and Python without bazel but with cmake and FindTensorFlow.cmake

07 Feb 2018 440

https://github.com/mind/wheels

Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)

27 May 2017 886

https://github.com/clara-parabricks/GenomeWorks

SDK for GPU accelerated genome assembly and analysis

31 May 2019 284

https://github.com/neoheartbeats/neoheartbeats-kernel

An architecture for LLMs' continual-learning and long-term memories

Open3D-build

Provide Docker build sequences of Open3D for various environments.

https://github.com/NVIDIA/nvImageCodec

A nvImageCodec library of GPU- and CPU- accelerated codecs featuring a unified interface

watsor

Object detection for video surveillance

20 Jun 2020 244

PyTorch-build

Provide Docker build sequences of PyTorch for various environments.