Learn practical skills, build real-world projects, and advance your career

Credits : This was a part of our Deep Learning Tutorial Series by our Prof. Subhashish Dasgupta at Praxis Business School Program.

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.optim import Adam, RMSprop
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
%matplotlib inline
from torch.utils.data import DataLoader, TensorDataset
# from torchvision.datasets import MNIST
# MNIST("D:/Notebooks/",download=True,)
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
device
device(type='cuda')