우분투에서 docker를 활용한 인공지능 개발환경 만들기

시스템 환경

• OS : Ubuntu 20.04
• graphics
  • GTX 970
  • GTX 1080 ti
• ram : 16GB
• storage
  • ssd 500GB
  • hdd 1TB

그래픽 드라이버 설치

그래픽 카드 확인

lspci | grep -i VGA

드라이버 설치

sudo apt-get install ubuntu-drivers-common
sudo ubuntu-drivers autoinstall

설치 확인

nvidia-smi

Cuda/Cudnn 설치

Ubuntu 20.04에 CUDA Toolkit와 cuDNN, Tensorflow 설치하기

cuda 설치 방법

Cuda 설치

TensorFlow 2.1.0이상은 CUDA 10.1을 지원하므로 CUDA 10.1을 설치한다.

# gcc 버전 바꾸기
sudo update-alternatives --config gcc