[linux] tensorflow-gpu 사용하기

1. Nvidia Driver 설치

Nvidia Driver 설치

ubuntu-drivers devices
sudo apt-get install nvidia-driver-470

Nvidia Driver 설치 확인

아래 명령어를 통해 GPU 상태 정보와 Driver 정보를 확인 가능하다.

nvidia-smi

faild 라고 뜬다면 재부팅 후 다시 시도한다

sudo reboot

2. CUDA 설치

https://developer.nvidia.com/cuda-toolkit-archive 접속

Latest Release 버전을 선택하고

자신에게 맞는 운영체제를 선택하면 아래와 같이 선택하면 설치 명령어가 뜬다.

https://velog.velcdn.com/images%2Fsomnode%2Fpost%2Fcafeaa50-78a8-4708-b83f-947f6f7044fd%2Fimage.png

아래 적혀있는 명령어를 그대로 작성해준다.

Untitled

설치 완료 후 reboot을 해준다.