pip install pytorch gpu code example
Example 1: torch conda
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
Example 2: mac install pytorch 3.6
conda install pytorch torchvision -c pytorch
Example 3: install pytorch cuda 10
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch
Example 4: torch import
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader
from torchvision import datasets, transforms, models
from torchvision.utils import make_grid
import numpy as np
import pandas as pd
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
%matplotlib inline
transform = transforms.ToTensor()
train_data = datasets.FashionMNIST(root='../Data', train=True, download=True, transform=transform)
test_data = datasets.FashionMNIST(root='../Data', train=False, download=True, transform=transform)