AI | ML/AI 개발 | CUDA

[CUDA] 윈도우 환경에서 CUDA 버전 변경

깜태 2021. 7. 9. 16:19
728x90

어떤 github의 딥러닝 코드는 tensorflow 1.14 버전을 썼는데, 1.14 버전은 CUDA 11버전이 호환되지 않아서

 

1. python 버전 변경

2. tensorflow 재설치

3. CUDA 버전 번경 (11 -> 10)

 

의 절차가 필요하였다.

 

나의 상황은 다음과 같고, 하나씩 적용해봤다.

 

GPU : 2060 RTX

OS : Windows 10

CUDA : 11.0

Python : 3.8

 

1. Python 버전 변경

다행히도 Anaconda를 쓰고 있어서, conda 환경으로 3.6 을 새로 만들어서 쉽게 해결하였다.

 

2. tensorflow 재설치

마찬가지로, conda 환경으로 새로 설치하고 나니 pip install tensorflow-gpu==1.14.0 으로 쉽게 해결하였다.

 

3. CUDA 버전 변경

1) CUDA 10 버전으로 바꾸기 위해 CUDA 10 을 설치한다. 

https://developer.nvidia.com/cuda-10.0-download-archive

 

CUDA Toolkit 10.0 Archive

Select Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. Operating System Architecture Distribution Version Installer Type Do you want to cross-compile? Yes No Select Host Platform Click

developer.nvidia.com

 

2) 설치 시 사용자 고급으로 설정하여 cuda 관련 라이브러리만 설치한다.

3) CUDA 설치 후, 고급 시스템 설정 -> 환경 변수를 통해 CUDA 설정을 확인한다.

4) CUDA_PATH 를 이용하고자 하는 버전으로 변경하고, Path 내에서도 사용하고 싶은 버전을 상위로 올린다.

5) 설정이 완료되면 cmd 창으로 nvcc -V 명령어로 cuda 버전이 몇으로 나오는지 확인한다.

6) 혹시나 안 되면, 재부팅하여 확인한다.

 

참고 ) https://www.programmersought.com/article/60413342978/

728x90