1. 설치할 CUDA, cuDNN 버전 확인
tensorflow를 사용하실 분은 CUDA도 설치해줍니다.
텐서플로우 홈페이지에서 본인의 작업 환경에 맞는 CUDA 버전을 확인하고 설치해주시면 좋습니다.
소스에서 빌드 | TensorFlow
소스에서 빌드 소스에서 TensorFlow pip 패키지를 빌드하고 Ubuntu Linux 및 macOS에 설치합니다. 명령어는 다른 시스템에도 적용될 수 있지만, Ubuntu 및 macOS용으로만 테스트되었으며 지원됩니다.참고: 잘
www.tensorflow.org

2. CUDA , cuDNN 설치
아래 사이트에서 설치할 CUDA Toolkit을 선택해 설치해봅시다.
CUDA Toolkit Archive
Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production
developer.nvidia.com

CUDA와 Cudnn 설치는 아래 블로그를 참고하였습니다.
Ubuntu 20.04에 CUDA Toolkit 11.2, cuDNN 8.1.0, Tensorflow 설치하기
Ubuntu 20.04에 CUDA Toolkit와 cuDNN, Tensorflow를 설치하는 방법을 다룹니다. 텐서플로우 최신 버전( 현재 2.5.0 )에서 CUDA 11을 사용하고 있어 수정했습니다. 2020. 7. 19 최초 작성 2020. 8. 31 2021. 5...
webnautes.tistory.com
'공부공부공부를 합시다 > 이것저것 설치해 봅시다' 카테고리의 다른 글
맥 OS 한영 전환키 설정 (0) | 2022.11.28 |
---|---|
[Ubuntu/Jupyter 서버 구축] Jupyter Notebook 설치 및 설정하기 [4] (0) | 2022.08.10 |
[Ubuntu/Jupyter 서버 구축] Nvidia 설치 [2] (0) | 2022.08.10 |
[Ubuntu/Jupyter 서버 구축] Ubuntu20.04 기초 설정 [1] (0) | 2022.07.27 |
[Elastic] 엘라스틱서치(ElasticSearch) 설치하기 (0) | 2022.05.09 |
댓글