시스템 구성

CUDA Toolkit 설치

추가 드라이버 탭에서 NVIDIA driver metapackage를 nvidia-driver-440에서 사용합니다 (프로플래터)를 선택하고 변경 사항 적용 버튼을 누릅니다.

$ sudo apt install nvidia-cuda-toolkit

Anaconda 설치

$ sh ./Anaconda3-2020.02-Linux-x86_64.sh

Tensorflow, Pytorch, CuPy 설치 및 동작 확인

  $ conda create --name=tf1 python=3.7 jupyter cudatoolkit==10.1.243
  $ conda activate tf1
  $ conda install tensorflow-gpu==1.14.0
  $ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
  $ pip install cupy-cuda101 chainer chainercv

  $ conda create --name=tf2 jupyter cudatoolkit==10.1.243
  $ conda activate tf2
  $ conda install tensorflow-gpu
  $ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
  $ pip install cupy-cuda101 chainer chainercv

GPU의 동작 확인

  $ python -c 'from tensorflow.python.client import device_lib; device_lib.list_local_devices()'

  $ python -c 'import torch; print(torch.cuda.is_available())'

  $ python -c 'import chainer; chainer.print_runtime_info()'

기타

$ sudo ubuntu-drivers devices

$ nvidia-smi