NVIDIA CUDA toolkit on Ubuntu20.04

システム構成

• PC：DELL XPS15 2017（GeForce GTX 1050）
• OS：Ubuntu Linux 20.04 LTS Desktop 日本語版

CUDA Toolkitインストール

このインストール方法は Anaconda の Python を利用しないと使えないかもしれないです。
conda install でAnaconda で用意されている CUDA と cuDNN をインストールしているように見えます。

1. Ubuntu のログインメニューから「ソフトウェアのアップデート」を開く

2. 「追加のドライバー」タブで「NVIDIA driver metapackageをnvidia-driver-440から使用します（プロプラエタリ）」を選択して、「変更の適用」ボタンを押下

   

3. CUDA Toolkit のインストール

   $ sudo apt install nvidia-cuda-toolkit
   

Anacondaのインストール

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

Tensorflow、Pytorch、CuPyのインストールと動作確認

Tensorflow1系と2系は同居できないので、両方を使いたい場合は仮想環境を作る必要があります。
あと、よくわかりませんが、python3.8 に tensorflow1.14 はインストールできないようです。

• Tensorflow1系

  $ 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

• Tensorflow2系

  $ 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の動作確認

• Tensorflow

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

• Pytorch

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

• CuPy

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

その他

1. Ubuntu がGPUを認識しているかの確認

   $ sudo ubuntu-drivers devices
   

2. GPUドライバが正しくインストールされているかの確認

   $ nvidia-smi