#1 소소의 소개
인공지능을 공부하고 싶어 그동안의 버킷리스트이던 블로그를 개설했습니다.
실력은 아직 좋지 않지만, 알고리즘 공부와 논문 스터디를 위주로 공부하여 정리할 예정입니다.
#2 개발 환경
- 그래픽 카드 : RTX2060
- 그래픽 드라이브 : 432.00
- 운영체제 : 윈도우10 64bit
- PYTHON : 3.5.2
- CUDA : 10.0
- cudnn : 7.6.5
- tensorflow-gpu : 2.0.0
- pytorch : 1.2.0, torchvision : 0.4.0
#3 tensorflow-gpu 설치
1) RTX2060은 compute compatibility가 7.5 (https://developer.nvidia.com/cuda-gpus#collapse2)
2) 장치관리자에서 내 그래픽 드라이버 버전 확인 : 432.00
3) https://www.tensorflow.org/install/gpu 에서 CUDA 10.0 requires 410.x or higher.라고 라고 했으니 CUDA 10.0 설치
- https://developer.nvidia.com/cuda-toolkit-archive에서 CUDA10.0 다운로드
- nvcc --version으로 설치 확인
3) Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.0 다운로드 후 압축 해제
4) 환경 변수 속 cuda path에 압축해제한 세 개 폴더 복사, 붙여넣기
5) pip install tensorflow-gpu
- 2.0.0 버전이 깔렸는데, 잘 안될 경우 추후에 1.13.X로 다운그레이드할 예정
(import tensorflow는 문제없음)
※ 참고 : https://pythonkim.tistory.com/137
#4 pytorch 설치
1) https://pytorch.org/ > Get Started
2) 내 cuda 버전은 10.0인데, 9.2, 10.1, None 선택지 밖에 없어서 아래 방식으로 설치
- https://download.pytorch.org/whl/torch_stable.html 에서 아래 두 파일 다운로드 후 pip install [ 두 파일 ] 실행
① cu100/torch-1.2.0-cp35-cp35m-win_amd64.whl
② cu100/torchvision-0.4.0-cp35-cp35m-win_amd64.whl
3) python에서 import torch, import torchvision 확인