CUDA, cuDNNをインストールする
Ubuntu20.04でCUDA10.1を使いたい場合、公式サイトではUbuntu18.04が出ているがそれをそのまま使ってインストールすることが可能。
CUDA10.1をインストールする
以下のサイトへ行きインストールを指示通りに行う。
CUDA Toolkit 10.1 Original Archive
Get CUDA Toolkit 10.1 for Windows, Linux, and Mac OSX.
developer.nvidia.com
「cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb」をダウンロードし、以下のコマンドを実行
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-10.1cuDNN7をインストールする
以下のサイトから「cuDNN Library for Linux」をダウンロードする。そして以下のコマンドで解凍。
tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgzcuDNN Archive
cuDNN Archive
Download releases from the GPU-accelerated primitive library for deep neural networks.
developer.nvidia.com
そしてcuda-10.1へコピー作業を行う。
# cuda-10.1にコピーする場合
sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn*.h /usr/local/cuda-10.1/lib64/libcudnn*参考
Ubuntu 20.04へのCUDAインストール方法 - Qiita
Ubuntu 20.04 に NVIDIA の CUDA のインストール方法です。私は主に Google Cloud や Amazon EC2 の GPU インスタンスを使っているので、それ前提で書…
qiita.com