파이토치(pytouch), 텐서플로우(tensorflow)를 위한 쿠다(cuda), cuDNN 설치
- IT/IT 프로그램 설치
- 2025. 2. 27. 10:58
<준비사항>
- 내 그래픽카드 확인
- 텐서플로우(tensorflow), 파이토치(pytouch) 중 무엇을 쓸 것인지 결정(둘 중 하나만 쓸거면, 해당 사이트에서 지원하는 최신버전 확인 후 설치하면 됨)
- 둘 다 쓸 경우에는 텐서플로우(tensorflow), 파이토치(pytouch) 둘 다 지원하는 쿠다(cuda), cuDNN을 설치해야 함(잘못 설치할 경우 포맷해야되는 경우가 있음)
- 가상환경을 구분하여 설치하는 것을 권장(텐서와 토치가 동시에 설치안된다는 글도 있어, 확인이 필요함)
- 내 PC 그래픽카드 확인(장치관리자 활용 및 내 PC 사양 확인 툴)
- 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 버전)
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
쿠다 설치 방법은 다음 사이트에서 확인 필요
'IT > IT 프로그램 설치' 카테고리의 다른 글
[오라클 설치] 오라클 XE 설치 및 다운로드 (0) | 2018.09.03 |
---|---|
[MariaDB 설치] mariadb 설치 방법 및 테스트 (0) | 2018.08.29 |