Skip to content

Instantly share code, notes, and snippets.

@dusty-nv
dusty-nv / pytorch_jetson_install.sh
Last active October 21, 2021 01:28
Install procedure for pyTorch on NVIDIA Jetson TX1/TX2 with JetPack <= 3.2.1. For JetPack 4.2 and Xavier/Nano/TX2, see https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/
#!/bin/bash
#
# EDIT: this script is outdated, please see https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-6-0-now-available
#
sudo apt-get install python-pip
# upgrade pip
pip install -U pip
pip --version
# pip 9.0.1 from /home/ubuntu/.local/lib/python2.7/site-packages (python 2.7)
@dusty-nv
dusty-nv / nano-read-mac.sh
Created June 7, 2019 19:49
Tool to read the real MAC address of a Jetson Nano
#!/bin/bash
#
# This script reads the correct MAC address of a Jetson Nano from the onboard EEPROM over i2c.
# First install i2c-tools package. Then make this script executable, and run it with sudo:
#
# $ sudo apt-get install i2c-tools
# $ chmod +x nano-read-mac.sh
# $ sudo ./nano-read-mac.sh
# MAC address is xx:xx:xx:xx:xx:xx
#
@dusty-nv
dusty-nv / README-vnc.txt
Created September 6, 2019 18:33
Configuring VNC from the command-line (L4T R32.2)
=======================================================================
README-vnc
Linux for Tegra
Configuring VNC from the command-line
=======================================================================
A VNC server allows access to the graphical display of a Linux for Tegra system
over the network. This allows you to work physically remote from the Linux for
Tegra system, and avoids the need to connect an HDMI display, USB keyboard, or
mouse.
@dusty-nv
dusty-nv / pytorch-diff-jetpack-4.2.patch
Last active July 8, 2020 01:48
PyTorch patch for building on JetPack 4.2 / 4.3
diff --git a/aten/src/ATen/cuda/CUDAContext.cpp b/aten/src/ATen/cuda/CUDAContext.cpp
index e48c020b03..0ecc111c4b 100644
--- a/aten/src/ATen/cuda/CUDAContext.cpp
+++ b/aten/src/ATen/cuda/CUDAContext.cpp
@@ -24,6 +24,8 @@ void initCUDAContextVectors() {
void initDeviceProperty(DeviceIndex device_index) {
cudaDeviceProp device_prop;
AT_CUDA_CHECK(cudaGetDeviceProperties(&device_prop, device_index));
+ // patch for "too many resources requested for launch"
+ device_prop.maxThreadsPerBlock = device_prop.maxThreadsPerBlock / 2;
@dusty-nv
dusty-nv / pytorch-1.10-jetpack-4.5.1.patch
Last active April 15, 2024 18:31
PyTorch patch for building on JetPack >= 4.4
diff --git a/aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h b/aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h
index 2aac442d21..f2321dad7a 100644
--- a/aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h
+++ b/aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h
@@ -26,6 +26,9 @@ namespace {
// Most likely we will do aarch32 support with inline asm.
#if defined(__aarch64__)
+// See https://github.com/pytorch/pytorch/issues/47098
+#if defined(__clang__) || (__GNUC__ > 8 || (__GNUC__ == 8 && __GNUC_MINOR__ > 3))
@dusty-nv
dusty-nv / labels.json
Created June 10, 2021 19:31
NLU-Evaluation-Data intent/slot labels
"intent_labels": [
"alarm_query",
"alarm_remove",
"alarm_set",
"audio_volume_down",
"audio_volume_mute",
"audio_volume_up",
"calendar_query",
"calendar_remove",
"calendar_set",
@dusty-nv
dusty-nv / kinetics_moments_labels.txt
Created March 18, 2023 04:57
Class labels for action recognition models trained on Kinetics-700 + Moments In Time datasets
slicing onion
jetskiing
building cabinet
eating hotdog
building lego
jumping jacks
kissing
bee keeping
blowing glass
curling eyelashes