ratsgo / embedding

한국어 임베딩 (Sentence Embeddings Using Korean Corpora)
https://ratsgo.github.io/embedding
MIT License
452 stars 129 forks source link

tensorflow version을 업데이트 하려면? #118

Closed thegreatkingbear closed 3 years ago

thegreatkingbear commented 3 years ago

1060과 1080에서는 잘 돌렸는데, 이번에 3090으로 업그레이드를 한 후 같은 bert tuning 과제를 돌려보니, failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED 라는 에러가 납니다. 구글링 해보니 두 가지 해결책이 나오는데,

  1. cuda toolkit version 9.0의 patch 네 개를 해준다.
  2. 텐서플로 버전을 1.14로 업데이트해준다. 여기서 1번은 했는데, 2번은 막상 하려니 이때까지 아나콘다 등의 환경에서만 텐서플로를 깔아본지라 어떻게 해야할지 감이 안 오네요 ;