[python] pytorch 라이브러리 설치하기

파이썬에서 pytorch 를 사용하고자 하는데 필요한 라이브러리 설치 명령어와 import 를 정리해 두었다. 

 

 

pip install opencv-python
conda install pytorch torchvision -c pytorch
conda install -c conda-forge imgaug
conda install albumentations -c conda-forge

 

 

from PIL import Image
import os
from pathlib import Path
import shutil
import cv2
import numpy as np
import time
from tqdm import tqdm
import torch
import torchvision
from torch.utils.data import Dataset
from torchvision import transforms
import albumentations
from albumentations.pytorch.transforms import ToTensor
from matplotlib import pyplot as plt
import matplotlib.image as img