機械学習のために我に Cuda を...

はじめに

Nvidia-smi をインストールしたので、 naya-hira.hatenablog.com 次に必要になるのはもちろんCudaですね。

今回の環境

OS : Ubuntu 18.04
GPU : RTX 2070 Super
Nvidia-smi Cuda version : 10.2

2020/05/08 現在の情報で、古くなっていたり、リンクが切れている可能性があります。

Tookit をインストールする

developer.nvidia.com

以上のリンクから、以下の画像のように自身の環境通りに選択していきます。
f:id:naya_hira:20200508224147p:plain
選択するとその下部にコマンドが表示されるので下記のように実行していきます。 `自身の環境により変化します

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

パスを通す

~/.bashrc に下記を追記します。

#cuda-xxx <- version は自身の環境に合わせる必要があります
export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

設定を再読み込みする source ~/.bashrc

動作確認

動作確認とバージョン確認
nvcc -V

#今回の実行結果
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

終わりに

そろそろ機械学習したいんですが....

naya_hira’s diary

三日坊主の備忘録