사용자 도구

사이트 도구


wiki:ai:python:tensorflow

문서의 이전 판입니다!


tensorflow

  • description : tensorflow 관련 내용 기술
  • author : 도봉산핵주먹
  • email : hylee@repia.com
  • lastupdate : 2020-07-08

Tensorflow

Install

Ubuntu 18.04 (GPU version)

1. CUDA Toolkit 10.1 update2

 https://developer.nvidia.com/cuda-toolkit-archive (회원 가입 필요 없음)

tip:위 버전이 Tensorflow2 gpu 호환 가능 버전

2. cuDNN v7.6.4 설치

 https://developer.nvidia.com/rdp/cudnn-archive (회원가입 필요)

tip:위 버전이 CUDA Toolkit 10.1 update2 호환 버전

3. python3.8 설치

 https://linuxize.com/post/how-to-install-python-3-8-on-ubuntu-18-04/

4. NVIDIA driver 설치

$> nvidia-smi
$> ubuntu-drivers devices
$> sudo ubuntu-drivers autoinstall

https://linuxconfig.org/how-to-install-the-nvidia-drivers-on-ubuntu-18-04-bionic-beaver-linux
우분투에서 NVIDIA 드라이버 설치 방법

gpu 사용 확인 code

from tensorflow.python.client import device_lib
device_lib.list_local_devices()


Ref

Tip

Troubleshooting

Ref

/volume1/web/dokuwiki/data/attic/wiki/ai/python/tensorflow.1597749051.txt.gz · 마지막으로 수정됨: 2022/03/10 19:52 (바깥 편집)