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

 

共通編

NVIDIAドライバ インストール

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

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

Visual Studio インストール

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

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

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

Release Notes :: CUDA Toolkit Documentation
The Release Notes for the CUDA Toolkit.

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

 

 

 

Pytorch編

 

PyTorch CUDA バージョンの確認

PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.

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

CUDA Toolkit インストール

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 ...

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

cuDNN ダウンロード

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″にコピペすれば終了.(↓参考)

CUDA cuDNNのインストール手順
インストール環境 今回インストールしていくPC環境: OS Windows 10 Pro CPU Intel Core...

 

 

 

Tensorflow編

Tensorflow CUDA バージョンの確認

Build from source on Windows  |  TensorFlow

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

CUDA Toolkit インストール

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 ...

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

cuDNN ダウンロード

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″にコピペすれば終了.(↓参考)

CUDA cuDNNのインストール手順
インストール環境 今回インストールしていくPC環境: OS Windows 10 Pro CPU Intel Core...