TensorFlow/PyTorchのNVIDIA GPU動作実績組み合わせ

Environment

概要

  • TensorFlow/PyTorch＋NVIDIA GPUで学習環境を構築したときの、動作実績ありの組み合わせ(環境構築の覚え書き)

TensorFlow

  • TensorFlow1.15

    • CUDA：10.0
    • cuDNN：7.6.5

  • TensorFlow2.10

    • CUDA：11.8
    • cuDNN：8.6.0

PyTorch

  • PyTorch1.12.1
    • CUDA：10.2
    • cuDNN：8.5.0

補足

  • NVIDIA GPUのドライバは最新にしておいたほうがよいと思われる
  • インストールは、どれも以下手順の実行のみで、GPU認識までできている(複数の組み合わせのインストールも可)
    1. NVIDIA GPUドライバのインストール(インストーラー実行)
    2. CUDAのインストール(インストーラー実行)
    3. cuDNNのインストール(インストーラー実行)
    4. TensorFlow/PyTorchのインストール(Pythonの仮想環境上でインストール)