소개

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의 모델을 실행하게 되어 실제로 동작하는 것을 확인하면 된다.