Skip to content

Instantly share code, notes, and snippets.

View imadelh's full-sized avatar
Cooking stuff

Imad E. imadelh

Cooking stuff
View GitHub Profile
imadelh /
Last active June 19, 2022 15:28
Nvidia Docker - Jetson Nano
# Copied and modified from
# Copyright (c) 2017, Technica Corporation. All rights reserved.
NV_LIBS="/usr/lib/aarch64-linux-gnu \
/usr/local/cuda/lib64 \
/usr/local/cuda \
/usr/src/tensorrt \
/usr/local/cuda-10.0 \
/usr/include \
imadelh /
Created July 28, 2019 10:11
Training CNN
import torch
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import logging
import numpy as np
# Define transformers