【備忘録】PyTorchでGPUを認識させる|Anaconda

 

NVIDIAドライバ インストール

NVIDIA
Download drivers for NVIDIA products including GeForce graphics cards, nForce motherboards, Quadro workstations, and more. Update your graphics card drivers to...

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

 

 

Visual Studio インストール

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

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

 

 

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 インストール

Release Notes :: CUDA Toolkit Documentation
The Release Notes for the 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 ...

▼ CUDA10.2/CUDA11.1で,自分の使っているGPUのドライバーのバージョンの両方を満たすCUDAをインストールする.

 

 

cuDNN ダウンロード

cuDNN Archive
NVIDIA cuDNN is a GPU-accelerated library of primitives 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...