새소식

개발환경세팅

[draft] Nvidia Quadro RTX 4000 딥러닝 개발환경 세팅기

  • -

설치순서

1. 그래픽카드 최신 드라이버 깔기 (nvidia 홈페이지)

2. visual studio 2019 (community version)

3. CUDA 11.1   (_456.43 _ win10) 설치

4. cuDNN-v8.0.5 (windows x64) >> 압축해제 후 CUDA 설치 디렉토리에 맞춰 파일 복사

 

** CUDA 설치 디렉토리(기본) : Program Files / NVIDIA GPU Computing Toolkit / CUDA 

 

 

 

A. tensorflow 테스트

 

<tensorflow-gpu 2.4.0 version>

 

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

다음 코드 실행해보니

 

대략...

1. GPU는 확인됨

2. 근데 라이브러리가 딴건 다 있는데 cusolver64_10.dll이 없어서 에러가 난다는 말.

3. 그래서 GPU 등록을 skip한다는거

 

 

그래서 문제의 그 라이브러리를 보기 위해, CUDA 설치 폴더의 bin 디렉토리를 보았다!

cusolver64_11.dll이 있다. (10이 아니라)
그래서 바꿔보았다!

 

그리고 다시 실행.

 

 

만세! 잘된다.

 

 

 

 

B. PyTorch Test

 

(아나콘다 가상환경에서 시작)

 

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

패키지 쪼끔 더 설치

conda install jupyter notebook ipython

 

그리고 코드 테스트~

 

 

잘된다 잘된다~~

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.