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
2) 설치 시 사용자 고급으로 설정하여 cuda 관련 라이브러리만 설치한다.
3) CUDA 설치 후, 고급 시스템 설정 -> 환경 변수를 통해 CUDA 설정을 확인한다.
4) CUDA_PATH 를 이용하고자 하는 버전으로 변경하고, Path 내에서도 사용하고 싶은 버전을 상위로 올린다.
5) 설정이 완료되면 cmd 창으로 nvcc -V 명령어로 cuda 버전이 몇으로 나오는지 확인한다.
6) 혹시나 안 되면, 재부팅하여 확인한다.
728x90
'AI | ML > AI 개발 | CUDA' 카테고리의 다른 글
[C++] onnx 결과값 Mat 타입으로 변경 (0) | 2021.11.30 |
---|---|
[Pytorch Lightning] Pytorch-Lightning 사용해보기 (0) | 2021.07.21 |
[CUDA] CUDA 드라이버 호환 확인 (0) | 2020.12.21 |
[TensorRT] Window 환경에서 PyTorch로 TensorRT 사용하기 (4) | 2019.10.07 |
Window C++ ONNX 환경 실행 (0) | 2019.09.24 |