본문 바로가기

Linux

[환경설정(re)] Nvidia drvier & CUDA & CuDNN + tensorflow 버전 확인

CUDA 삭제 후 재설치

    • run file로 설치함
    • cuda path 설정

  • sudo ldconfig 입력
  • cuda 삭제 방법(나중에 지울때 참고) : To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-11.6/bin

 

11.6 version을 설치했는데 nvcc --version으로 버전확인 해본 결과, 이전에 깔았던 10.1 버전이 출력되었다.

해결 방법 >> PATH 설정

CuDNN 설치

https://cafepurple.tistory.com/39

 

Ubuntu 18.04 에서 CUDA+CUDNN 설치하기

Ubuntu 18.04에 CUDA+CUDNN를 설치해 보도록 하겠습니다. OpenCV까지 설치할 계획으로, 하나의 포스팅에서 전부 다루기에는 내용이 길어서 글을 나누어 작성하겠습니다. 저도 리눅스에 설치하는 것은 처

cafepurple.tistory.com

  • CUDA 11.6 버전을 설치하였으므로, 맞는 version 확인하고, CuDNN은 8.4.0v 설치
  • cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
  • https://driz2le.tistory.com/264
 

nvdia driver, CUDA, CUDNN 설치 여부 및 버전 검사

1. Nvidia Driver 설치 여부 및 버전 확인 $ cat /proc/driver/nvidia/version → 정상적으로 설치되어 있는 경우 아래와 같이 출력된다. NVRM version: NVIDIA UNIX x86_64 Kernel Module  430.34  W..

driz2le.tistory.com

tensorflow 버전확인

  • 터미널 창에 python 입력
import tensorflow as tf
print(tf.__version__)
  • 2.7.0 버전

최종

  • nvidia driver 510
  • cuda 11.6
  • cudnn 8.4.0
  • tensorflow 2.7.0
  • python 3.8.10
  • + bazel 5.2.0

수정

  • nvidia driver 515.48.07
  • cuda 10.1
  • cudnn 7605 (7.6?)
  • tensorflow 2.2
  • python 3.6.15