반응형
매번 환경구축이 귀찮아서 한번에 모아놓고 제가 보기위한 용으로 적어봅니다.
일자 : 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
반응형
'Python' 카테고리의 다른 글
무료웹드라이브(MediaFire) 사용하여 대용량 데이터셋 공유하기! 10GB까지 무료로 사용가능! (0) | 2020.03.17 |
---|---|
Tips)PyTorch 딥러닝(Deep Learining) 탬플릿을 사용해서 손쉽게 학습하기(fine-tuning) + 깃헙코드 O (0) | 2019.11.29 |
Tips)[Windows10] cuDNN, CUDA 재설치/설치확인 방법 (4) | 2019.11.07 |
댓글