CV・NLPハマりどころメモ

画像認識と自然言語処理を研究する中でうまくいかなかったことと、その対策をまとめる自分用メモが中心。

CUDA10.0ではなく10.1がインストールされてしまう[CUDA]

TensorflowがCUDA10.0にしか対応していないので,最新の10.1では無い方をインストールした時の話.

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=debnetwork

のベージからOSなどを選択した状態で出現する公式の手順にしてがってインストールしたところ,CUDA10.0ではなくCUDA10.1がインストールされてしまった.debパッケージも10.0のものを選択したにも関わらず.

調べてみると,どうやらインストールの際にバージョンを指定してあげないといけないらしい.

sudo dpkg -i cuda-repo-ubuntu1604_10.1.168-1_amd64.deb

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-10-0

上のコマンドを打つと,うまくいった.バージョンの指定の仕方がハイフンを使う独特な形式だ.