Setting, Error/Python, Python Library

[Install] RTX 3090, cuda 11.2 Tensorflow-gpu 설치

Juheon Kwak 2021. 11. 23. 19:02

※ 경험한 바로는 3090에 기본 Tensorflow-gpu를 설치하면 tensorflow가 gpu를 똑바로 잡지 못했다. (이유는 아직도 모르겠음)

 

[해결방법]

- tensorflow nightly 버전을 이용한다.

- nightly 버전 : 안정화된 tensorflow 버전이 아닌 매일매일 개발되고 있는 버전.

- 기존의 tensorflow와 똑같이 사용하면 된다.

- 아니면 tensorflow-gpu 2.4 버전을 사용한다. (다른 라이브러리들과 버전 충돌 주의)

 

$ pip3 install tf-nightly

 

$ pip3 install tensorflow-gpu==2.4.0

 

(참고)

https://pypi.org/project/tf-nightly/