파이토치(pytouch), 텐서플로우(tensorflow)를 위한 쿠다(cuda), cuDNN 설치

<준비사항>

 -  내 그래픽카드 확인

 -   텐서플로우(tensorflow), 파이토치(pytouch) 중 무엇을 쓸 것인지 결정(둘 중 하나만 쓸거면, 해당 사이트에서 지원하는 최신버전 확인 후 설치하면 됨)

 -   둘 다 쓸 경우에는 텐서플로우(tensorflow), 파이토치(pytouch) 둘 다 지원하는 쿠다(cuda), cuDNN을 설치해야 함(잘못 설치할 경우 포맷해야되는 경우가 있음)

 -   가상환경을 구분하여 설치하는 것을 권장(텐서와 토치가 동시에 설치안된다는 글도 있어, 확인이 필요함)

 

  1. 내 PC 그래픽카드 확인(장치관리자 활용 및 내 PC 사양 확인 툴)
  2. Tensorflow(텐서플로우) 버전과 Pytorch 버전을 확인 후 쿠다 버전 설치
    텐서플로우 사이트 : https://www.tensorflow.org/install/source_windows?hl=ko
 

Windows의 소스에서 빌드,Windows의 소스에서 빌드  |  TensorFlow

이 페이지는 Cloud Translation API를 통해 번역되었습니다. Windows의 소스에서 빌드,Windows의 소스에서 빌드 컬렉션을 사용해 정리하기 내 환경설정을 기준으로 콘텐츠를 저장하고 분류하세요. 소스에

www.tensorflow.org

파이토치 사이트 : https://pytorch.org/get-started/previous-versions/

 

Previous PyTorch Versions

Installing previous versions of PyTorch

pytorch.org

 

 

** 현재 나의 PC 환경에서는 cudatoolkot을 11.0 버전, cuDMM은 8.0버전으로 환경을 구성해야 텐서플로우와 파이토치 환경을 구성할 수 있을 것(다른 버전도 될 수도 있지만, 한번 오류나면 os 포맷해야되는 경우 발생하기 때문에 최대한 오류를 줄이는 것이 필요)

 

 

쿠다설치(11.0 버전)

https://developer.nvidia.com/cuda-11.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

 

CUDA Toolkit 11.0 Download

Get CUDA Toolkit 11.0 for Windows and Linux.

developer.nvidia.com

 

cuDNN 설치

https://developer.nvidia.com/cudnn

 

CUDA Deep Neural Network

cuDNN provides researchers and developers with high-performance GPU acceleration.

developer.nvidia.com

 

쿠다 설치 방법은 다음 사이트에서 확인 필요

https://lonaru-burnout.tistory.com/16

댓글

Designed by JB FACTORY