Usually, located at /usr/local/cuda/bin
$ nvprof python train_mnist.py
I prefer to use --print-gpu-trace.
-- Remove the history from | |
rm -rf .git | |
-- recreate the repos from the current content only | |
git init | |
git add . | |
git commit -m "Initial commit" | |
-- push to the github remote repos ensuring you overwrite history | |
git remote add origin [email protected]:<YOUR ACCOUNT>/<YOUR REPOS>.git |
import torch | |
from torch.utils.data import Dataset | |
from typing import Tuple | |
import numpy as np | |
import requests | |
import pandas as pd | |
#### LOADING THE MODEL | |
from torchvision.models import resnet18 |