Tensorflow-gpu(텐서플로), Pytorch(파이토치), Opencv(오픈씨비) 초간단 설치가이드.(tf_ver=1.13.1, pytorch=1.2.0, opencv(4.1.1))
본문 바로가기
Python

Tensorflow-gpu(텐서플로), Pytorch(파이토치), Opencv(오픈씨비) 초간단 설치가이드.(tf_ver=1.13.1, pytorch=1.2.0, opencv(4.1.1))

by Migos 2019. 10. 18.
반응형

매번 환경구축이 귀찮아서 한번에 모아놓고 제가 보기위한 용으로 적어봅니다.

 

일자 : 2019.10.18

RTX2080, CUDA 10.0, Cuda_toolkit 7.4.2

Tensorflow 2.0.0

pytorch 1.2.0

opencv 4.1.1.26(버젼 꼭 지킬 필요 없는듯)

 

[0] 설치된 CUDA 버젼 확인(선택사항)

nvcc--version

 

[1] Tensorflow-gpu 설치(1.13.1버젼으로 다운)

pip install tensorflow-gpu==1.13.1

 

[2] pytorch 설치

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

 

[3] opencv 설치

pip install opencv-python

 

설치 잘 되었나 확인

[1] Tensorflow-gpu

(tf-gpu) C:\Users\min>python

>>> import tensorflow as tf

>>> hello = tf.constant("hello")

>>> sess = tf.Session()

>>> print(sess.run(hello))

 

[2] pytorch

(tf-gpu) C:\Users\min>python

>>>import torch

>>>torch.cuda.get_device_name(0)

>>>torch.cuda.is_available()

>>>print(torch.__version__)

[3] open-cv

(tf-gpu) C:\Users\min>python

>>>import cv2

반응형

댓글