CUDAとcuDNNとPyTorchとPythonのバージョンの対応確認
※バージョンの対応がすべて揃わないとディープラーニングの練習環境が整わない。
インストール成功例
確認環境
Windows 10 64 bit
Nvidia GeForce GTX1070 ドライバ 398.75
CUDA
cuda_9.2.148.1_windows
cudnn
cudnn-9.1-windows10-x64-v7
Anaconda navigator からインストール
pytorch04gpu
Pytorch
Installing previous versions of PyTorch ページから
# CUDA 9.2 pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
上記を使用する。
公式サイトの Get started からインストールできるバージョンが torch==1.3.1 なので他とバージョンが合わなかった。
Python のバージョン
3.69