관리 메뉴

미래기술연구소

Tensorflow 설치 본문

카테고리 없음

Tensorflow 설치

I s a a c 2020. 9. 27. 21:06
728x90
반응형

 

python -m pip install --upgrade pip

pip install tensorflow-gpu 또는

pip install --ignore-installed --upgrade tensorflow-gpu 을 입력한다

 

Tensorflow 를 지원하는 NVIDIA 그래픽 카드를 사용한다면 CUDA를 설치해야 합니다.

- NVIDIA Cuda Toolkit 설치: https://developer.nvidia.com/cuda-toolkit-archive

또한 cuDNN을 설치해야 합니다. cuDNN을 받으려면 NVIDIA DEVELOPER 홈페이지에 가입이 필요합니다. 텐서플로우(Tensorflow) GPU 버전 설치하기는 이곳에서 다루지 않겠습니다.

- cuDNN 다운로드: https://developer.nvidia.com/rdp/cudnn-archive

물론 tensorflow를 설치할 때 conda를 사용해도 됩니다

 

 

 

728x90
반응형