[Ubuntu] 우분투 Tensorflow GPU사용 가능한 버전 설치

JeongHyun·2024년 3월 8일

Python

목록 보기
2/3

1. Tensorflow-2.15.0

우분투에서 Tensorflow가 필요한 상황이 생겨서 설치해보려고 한다.
예전에 최신버전으로 막 설치했다가 Tensorflow가 gpu를 못찾는 경우가 종종 있어서
버전에 맞춰 설치하겠다.

인공지능 학습에 필요한 환경인 : cuda 11.8 cndnn 8.7.0 설치는 예전에 포스팅 했기에 생략하겠습니다.

cuda 11.8 cndnn 8.7.0 개발환경 설치

1. 링크에서 원하는 버전, 환경의 텐서플로어를 다운받습니다.

https://www.tensorflow.org/install/pip?hl=ko#package_location

2. 다운받은 폴더 위치에서 terminal 창으로 명령어로 설치를 진행합니다.

저는 python 3.11버전 gpu지원을 다운받겠습니다. 원하는 걸 받아서 파일 이름대로 설치해 주세요.

$ pip install tensorflow-2.15.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

3. 설치 확인

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


GPU가 잡히면 성공~!

profile
하루하루 의미 있게

0개의 댓글