初めに

　今回がQiita初投稿です。至らない点がありましたら教えていただけると嬉しいです。

この記事でやること

　NVIDIA Geforce GTX 1060が搭載されたPC上にpytorch,tensorflow,kerasの環境構築を行います。バージョンが違うと動かないことが多々あるので注意してください。

導入するもの

nvidia driver(バージョンは任意(?))
anaconda
python 3.6で
tensorflow-gpu 1.12.0
keras 2.2.4
pytorch 1.6.0
※anacondaからのみで使用する場合はcuda-toolkitなどはtensorflow-gpuなどをインストールするとき自動的に入れてくれるので自分で入れる必要はなさそうです。

動作環境

windows10
ubuntu20.04

nvidia driverのインストール(ubuntuの場合)

参考

搭載されているGPUやドライバの確認

GPUの確認

ubuntu-drivers devices

2. ドライバのインストール

install_driver

sudo ubuntu-drivers install [ドライバのバージョン(435など recommendedされているものを使うと無難か?)]
#autoinstallはおすすめされていないらしい
#詳しくはubuntu-drivers --helpを実行する

3. PCを再起動する
4. nvidia-smiを実行しドライバなどの情報が取得できればOKです。

nvidia-smiが動かないとき

参考
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.
のようになることがある。この時はNouveauを無効化する

手順

sudo vim /etc/modprobe.d/blacklist-nouveau.confでファイルを作成し

blacklist-nouveau.conf

blacklist nouveau
options nouveau modeset=0

と記入
2. sudo update-initramfs -u
3. PCの再起動
詳細は上の参考を見てください。

nvidia driverのインストール(windowsの場合)

nvidia_driver_downloadから自分の使用している環境を選択しドライバをダウンロード
[]( - 自分の使用している環境はデスクトップで右クリックしnvidia コントロールパネルで調べられる??)
インストーラを起動する(すべてデフォルトのままでOKです)
コマンドプロンプトを開きnvidia-smiを実行しドライバなどの情報が取得できればOKです。

anacondaのインストール

python japanによるanacondaのインストールガイド
anaconda公式
に従ってインストールする

仮想環境の作成

anaconda promptを起動する(windowsの場合)。ubuntuではコマンドラインのところに(base)とついていればOKです。
参考
conda create -n 環境名 python=バージョン
で仮想環境を作成します。ここで環境名は任意、バージョンは3.6とします
環境のアクティベート
conda activate 環境名
コマンドラインのところに(環境名)とついていれば大丈夫です。

pytorchのインストール

pytorch公式で自分の環境を選択していくと必要なコマンドが出てきます。
私の環境では
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
でした。

gpuを認識できているのかの確認(pytorch)

参考(pytorch公式)
pythonを起動し

確認

import torch
torch.cuda.is_available()

を実行する。Trueなら認識できている

tensorflow-gpuのインストール

注意:cuda-toolkitがtensorflow-gpu==1.12.0ではバージョン9.1が求められるのでpytorchと違うものが必要になる。そのためpytorchと仮想環境を分ける必要がある
conda install tensorflow-gpu==1.12.0

gpuを認識できているのかの確認(tensorflow)

参考
pythonを起動し

確認

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

とする。GPUが表示されればGPUを認識できている

#kerasのインストール
conda install keras==2.2.4

今後の展望(未開設)

終わりに

ページに問題があれば教えていただけると嬉しいです。参考にさせていただいたページの皆様ありがとうございました。

参考一覧