Small Asteroid Blog

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

백엔드/Python

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

작은소행성☄️ 2020. 10. 29. 11:12
728x90

파이썬에서 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

 

 

 

728x90
반응형

'백엔드 > Python' 카테고리의 다른 글

('cv::adaptiveThreshold' Error) OpenCV(4.4.0) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-z5gfhs5a\opencv\modules\imgproc\src\thresh.cpp:1676: error: (-215:Assertion failed) src.type() == CV_8UC1 in function 'cv::adaptiveThreshold'  (0) 2020.12.03
[python] pip install albumentations error  (0) 2020.10.29
[python] No module named 'torch'  (0) 2020.10.29
[python] No module named 'cv2'  (0) 2020.10.29
[python] dataframe db에 저장  (0) 2020.10.07
'백엔드/Python' Related Articles more