imadelh

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

imadelh

#!/bin/bash
# Copied and modified from https://github.com/Technica-Corporation/Tegra-Docker
# 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 \