버그잡기
[머신러닝] Colab 에서 module 'tensorflow' has no attribute 'set_random_seed' 에러 날 때 텐서플로우 버전 낮추는 방법
_tech
2021. 1. 10. 19:35
728x90
반응형
코랩을 사용하면서 '모두의 딥러닝' 예제를 돌려보고 있는데,
이상하게도 자꾸 tensorflow 관련 에러가 날 때 해결했던 방법입니다.
찾아보니 텐서플로우 버전이 1.x일때와 2.x일 때에 사용 방법이 차이가 많이 나서 에러가 난다고 합니다.
그래서 코랩에서 텐서플로우 버전을 확인하고, 2.x 일 때 아래 버전으로 낮추는 방법에 대해서 설명하려고 합니다.
먼저 텐서플로우 버전 확인은 다음을 입력하여 확인 할 수 있습니다.
import tensorflow as tf
tf.__version__

확인해보시면 저처럼 2.x 버전으로 나오는 모습을 볼 수 있습니다.
이제 이 버전을 1.x 단계로 낮추기 위해서 아래의 명령어를 이용해 현재 사용하는 텐서플로우를 삭제하고 다시 깔아주도록 하겠습니다.
!pip uninstall tensorflow
!pip install tensorflow==1.15
명령어를 입력해주면 아래처럼 중간에 진행 확인을 묻는 부분이 나오는데 이 때 y를 입력해줍니다.

그러고 나서 좀 기다리면 재설치가 완료됩니다!
실행시키면 예제가 잘 돌아가는 모습입니다 :)

728x90
반응형