【備忘録】PyTorch/TensorflowでGPUを認識させる

共通編

NVIDIAドライバ インストール

www.nvidia.co.jp

最新の NVIDIA 公式ドライバーをダウンロード

https://www.nvidia.co.jp/Download/index.aspx?lang=jp

最新の NVIDIA 公式ドライバーをダウンロード

▼ ドライバーのバージョンを確認する．最新のドライバをインストールしていない人はインストールしておくといいかも．

Visual Studio インストール

docs.microsoft.com

Visual Studio 2019 バージョン 16.11 リリース ノート

https://docs.microsoft.com/ja-jp/visualstudio/releases/2019/release-notes

最新機能、バグの修正、および Visual Studio 2019 v16.11 のサポートを取得します。 今すぐダウンロード。

▼ Visual Studio ComminityからC++によるデスクトップ開発をインストールする．

GPUに対応しているCUDAのバージョンを確認

docs.nvidia.com

Release Notes :: CUDA Toolkit Documentation

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions__table-cuda-toolkit-driver-versions

The Release Notes for the CUDA Toolkit.

▼ 自分の使っているGPUのドライバーのバージョンに対応しているCUDAのバージョンを確認する．

Pytorch編

PyTorch CUDA バージョンの確認

pytorch.org

PyTorch

https://pytorch.org/get-started/locally/

▼ 現時点でPyTorchに対応しているのは"CUDA10.2"と"CUDA11.1"であることを確認する．(別バージョンでも大丈夫なことはありますが，念のため従っておきます．)

CUDA Toolkit インストール

NVIDIA Developer

CUDA Toolkit Archive

https://developer.nvidia.com/cuda-toolkit-archive

Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production drivers appropriate for your hardware configuration.

▼ 先ほど確認したGPUに対応しているCUDAをインストールする．今回はCUDA10.2/CUDA11.1のどちらかを選択する．

cuDNN ダウンロード

NVIDIA Developer

cuDNN Archive

https://developer.nvidia.com/rdp/cudnn-archive

Download releases from the GPU-accelerated primitive library for deep neural networks.

▼ インストールしたCUDAに対応するcuDNNをダウンロードする．(今回はWindowsで環境構築しているのでWindowsのものをダウンロード)

▼ ダウンロードしたファイルは"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1"にコピペすれば終了．(↓参考)

Tensorflow編

Tensorflow CUDA バージョンの確認

TensorFlow

Build from source on Windows  |  TensorFlow

https://www.tensorflow.org/install/source_windows

▼ 上記サイトからGPUに対応したCUDAのバージョンとcuDNNのバージョンを確認する．また．環境に対応したtensorflow-gpuをインストールする．

CUDA Toolkit インストール

NVIDIA Developer

CUDA Toolkit Archive

https://developer.nvidia.com/cuda-toolkit-archive

Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production drivers appropriate for your hardware configuration.

▼ 先ほど確認したGPUに対応しているCUDAをインストールする．

cuDNN ダウンロード

NVIDIA Developer

cuDNN Archive

https://developer.nvidia.com/rdp/cudnn-archive

Download releases from the GPU-accelerated primitive library for deep neural networks.

▼ 先ほど確認したcuDNNの同じバージョンのものをインストールする．

▼ ダウンロードしたファイルは"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1"にコピペすれば終了．(↓参考)