Windows10で CUDA10.2をインストール

はじめに

Windows10にCUDAドライバを以前入れていたものの、TensorflowやPyTorchでうまく動作しなくなり、しばらくGPUなしで動かしていたが、いい加減バージョンアップしてみた。

きっかけ

ある日PyTorchでこんなエラーが。よし、上げよう。

RuntimeError: The NVIDIA driver on your system is too old (found version 10010). Please updat                           

e your GPU driver by downloading and installing a new version from the URL: http://www.nvidia                           

.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch versi                           

on that has been compiled with your version of the CUDA driver.                         

現状調査

そもそもインストールされているPyTorchのバージョンはというと、

pytorch                   1.7.0           py3.7_cuda110_cudnn8_0    pytorch                     
torchvision               0.8.1                py37_cu110    pytorch                        

なんかしらんが11用が入っている。

現状インストールされてCUDAは？

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver                   
Copyright (c) 2005-2017 NVIDIA Corporation                  
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017                 
Cuda compilation tools, release 9.0, V9.0.176                   

9.0だ。

WindowsでPyTorchを使うための最適なCUDAのバージョンは？

PyTorchのホームページを見てみる。これによると、
PyTorch1.7, CUDA 10.2の組み合わせがよさそう。よって10.2をインストールすることにする。

インストール

NVIDIAのホームページより、Windows10用の10.2のドライバをダウンロードしインストールする。

上記よりBase Insataller、Patch1, Patch2を順にダウンロード、インストールしていくだけ。

インストール後の確認

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:32:27_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.2, V10.2.89

10.2がインストールされている。

環境変数も以下の通り、10.2がCUDA PATHに設定されている。

CUDA_PATH='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2'                                
CUDA_PATH_V10_2='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2'                              
CUDA_PATH_V8_0='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0'                                
CUDA_PATH_V9_0='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0'                                
CUDA_PATH_V9_1='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1'                                

PyTorchのインストール

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

無事CUDA 10.2用のものがインストールされていることを確認。

$ conda list |grep torch
pytorch                   1.7.0           py3.7_cuda102_cudnn7_0    pytorch
torchaudio                0.7.0                      py37    pytorch
torchvision               0.8.1                py37_cu102    pytorch

この後はPyTorchのモデルを実行するなりして、実際に動作することを確認すればよい。