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
$ 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
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のモデルを実行するなりして、実際に動作することを確認すればよい。
Author And Source
この問題について(Windows10で CUDA10.2をインストール), 我々は、より多くの情報をここで見つけました https://qiita.com/kimisyo/items/c8107693b1aaba98b168著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .