2021/05/02 Updated by

CUDA

Windows 10へのインストール

[Up] Japanese English

CUDA Toolkit 11.1のインストール

  1. 現在の PyTorch の最新版 Stable(1.8.1) で Windows, conda に対応しているのは CUDA 10.2 と 11.1 である。
  2. CUDA Toolkit の 最新バージョン は 11.3 であるが、PyTorch Stable(1.8.1) が 11.1 を使うので CUDA Toolkit Archive から11.1.1 をダウンロードする。(Update version の代りに minor version が付くようになったようだ)
  3. ダウンロードしてきたexeファイル(cuda_11.1.1_456.81_win10.exe)を実行する。
  4. 「nVidiaのデバイスドライバをダウングレードする」と表示される。
  5. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\ にインストールされる。
  6. Nsight for Visual Studio 201[579] がインストールされる。
  7. Sampleがインストールされる。
    8. C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1
  8. Windowsの環境変数PATHに以下の2つのパスが自動で追加される。 
        C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp
  9. Windowsの環境変数に CUDA に関する値が自動で追加される。 
        CUDA_PATH               C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
    CUDA_PATH_V11_1         C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

    NVCUDASAMPLES_ROOT      C:\Program Files\NVIDIA Corporation\CUDA Samples\v11.1
    NVCUDASAMPLES11_1_ROOT  C:\Program Files\NVIDIA Corporation\CUDA Samples\v11.1

cuDNN のインストール

  1. nVidia の公式サイト https://developer.nvidia.com/rdp/cudnn-download にアクセスする。
  2. CUDAに適合した cuDNN 8.2系をダウンロードする。
    3.   cuDNN v8.2.0 [April 23rd, 2021], for CUDA 11.x
  cuDNN Library for Windows [x86]  ← [x86]という記述は間違いのようだ。ダウンロードされるのは cudnn-11.3-windows-x64-v8.2.0.53.zip なので。
  3. NVIDIA Deep Learning SDK Documentation
  4. cuDNN Installation Guide をよく読んで、指示に従ってインストールする。
  5. 3個のファイルを所定のフォルダにコピーする
    6. 展開して出来たフォルダのパスを PATH で表すと、次の通り。 
    copy PATH\cuda\bin\cudnn*.dll  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
copy PATH>\cuda\ include\cudnn*.h  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include
copy PATH>\cuda\lib\x64\cudnn*.lib  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
  6. 環境変数 CUDA_PATH を次の値に設定する。
    7.   変数名: CUDA_PATH 
  値:     C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
    最近は自動で設定されるようだ。また、バージョンが付加された変数名の値も自動で設定されるようだ。 
      変数名: CUDA_PATH_V11_0
  値:     C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0