[ TensorFlow ] 우분투 NVIDIA GPU CUDA® Toolkit 설치

[ TensorFlow ] 우분투 NVIDIA GPU CUDA® Toolkit 설치



1. NVIDIA CUDA 자료실 접속




2. 최신 CUDA Toolkit 다운로드

   - Download Latest CUDA Toolkit [클릭]



3. 운영체제 맞춰서 선택

    - Linux [클릭]



4. 운영 체제 맞춰서 선택

    - Linux [선택]
    - x86_64 [선택]
    - Ubuntu [선택]
    - 20.04 [선택]
    - runfile (local) [선택]



5. runfile (local) 선택 Base Installer 활성화

    - 터미널 [실행]



6. CUDA® Toolkit 다운로드

    $ wget https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux.run



7. CUDA® Toolkit  설치

    $ sudo sh cuda_11.5.1_495.29.05_linux.run



8. CUDA® Toolkit 약관에 동의 합니까?

    - accept [Enter]



9. CUDA® Installer ( Driver 는 GPU Driver 충돌이 있으니 제외) 

    - 설치 항목 [선택]
    - Install [Enter]



10. CUDA® Toolkit 설치 완료

    - 경고: 불완전한 설치! 이 설치는 CUDA 드라이버를 설치하지 않았습니다. [무시]



11. 환경 변수 추가


   
$ nano .profile [입력]



12. profile 마지막 추가

    > export PATH=$PATH:/usr/local/cuda-11.5/bin
    > export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
    - Ctrl + O [저장]
    - 저장 이름 확인 [Enter]
    - Ctrl + X [종료]



13. 재 부팅

    $ reboot [입력]



14. 환경 변수 확인

    $ printenv




15. CUDA 설치 확인

    $ nvcc --version





댓글