-
CUDA のインストール
CUDA ダウンロードリンク https://developer.nvidia.com/cuda-downloads -
CUDA のインストール
以下の手順はすべて ROOT 権限で操作します
chmod +x cuda_10.1.168_418.67_linux.run で.run ファイルを実行可能ファイルに変更
./ cuda_10.1.168_418.67_linux.run
CUDA をインストールします
CUDA をインストールすることを選択します。ここではグラフィックドライバーのインストールは選択していません。必要に応じて直接選択できます。
インストール完了
環境変数の設定
vim /etc/profile
CUDA 環境変数を追加
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
保存して終了後、環境変数を同期
source /etc/profile
nvcc -V
CUDA バージョンを確認し、CUDA バージョンが出力されれば、CUDA のインストールは成功です。
- cuDNN のインストール
cuDNN をダウンロード https://developer.nvidia.com/rdp/cudnn-archive
ダウンロード後、解凍し、解凍したファイルに移動
CUDA のソフトリンクを作成
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cuDNN7 のインストールが成功したか確認
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cuDNN8 のインストールが成功したか確認
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
CUDA+CUDNN のインストールが完了しました