AI [Tesorflow] GPU 설정 버전

[Tesorflow] GPU 설정 버전

Tensorflw gpu 버전을 설치할때 Cuda 와 Cudnn 에 대해 버전별로 맞추어야하는 상황이 자주발생하는데 Tensorflw 버전마다 아래와 같이 Cuda와 CuDnn 을 설치했던 내용을 정리한다. 

  1. Cuda 10.1 + CuDNN 7.6.5 (normally > 7.6) + TensorFlow 2.2.0 (TF >= 2.1 requires CUDA >=10.1)
  2. Cuda 10.1 + CuDNN 7.6.5 (normally > 7.6) + TensorFlow 2.1.0 (TF >= 2.1 requires CUDA >= 10.1)
  3. Cuda 10.0 + CuDNN 7.6.3 + / TensorFlow 1.13/1.14 / TensorFlow 2.0.
  4. Cuda 9.0 + CuDNN 7.0.5 + TensorFlow 1.10 works

Leave a Reply

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다

Related Post

[Pytorch] Pnemonia Prediction Model – Train[Pytorch] Pnemonia Prediction Model – Train

Pytorch를 이용하여 사내에서 학습을 해야하는 환경이 있어 GPU환경이 잘 설정되었는지 테스트겸 Kaggle에서 데이터셋을 다운로드 받고 학습을 돌렸다. 오랜만에 AI 쪽 업무를 하려니 많이 까먹기도 재밌어지기도 했다. 데이터셋 리뷰 https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia Kaggle의

인공신경망(Artificial Neural Network, ANN)인공신경망(Artificial Neural Network, ANN)

[ 구조 ] 딥러닝에서 가장 기본이 되는 개념은 바로 신경망(Neural Network)입니다. 신경망이란 인간의 뇌가 가지는 생물학적 특성 중 뉴런의 연결 구조를 가리키며, 이러한 신경망을 본떠 만든 네트워크 구조를 인공신경망(Artificial Neural