관리 메뉴

everymorning_

Tensorflow - 'Cuda Multi install' 본문

tensorflow

Tensorflow - 'Cuda Multi install'

dashhh 2019. 12. 14. 01:32

Tensorflow - 'Cuda Multi install'



Cuda Multi version Install

Tensorflow 버전에 따라 Cuda를 별도로 설정해 주어야 한다
Cuda는 한개만 설치 할 수 있는 개념으로 알고 있었으나, 자세히 보니 여러 버전을 동시에 설치 할 수 있는 것이었음.

Cuda를 여러개 설치 할 때 실수 한 점은 다음과 같다

  • Tensorflow 버전마다 맞는 Cuda 버전이 따로 있다는 사실
    • Cuda 버전이 안맞을 경우 gpu 연산에 사용하는 각종 라이브러리 호출에 실패하게 됨 (아래 사진)
      스크린샷, 2019-12-14 01-53-43
    • Cuda 버전을 맞출경우 아래와 같이 다시 잘 잡힌다.
      스크린샷, 2019-12-14 01-55-22

  • 해당 내용 상세는 아래 링크에서 확인 할 수 있다.
    • Tensorflow Version Dependency Index
      image
      image
    • 위 Index는 절대적인 기준은 아니지만 버전이 안맞으면 libcublas와 같은 library 호출을 실패한다 (해당 호출은 /usr/local/cuda-x.x 를 바라본다. 정확하게는 LD_LIBRARY_PATH를 쳐다보고 있다.)

CUDA 버전별로 손 쉽게 설치하는 방법
나중에 올려야겠네..

Comments