Tensorflow, CUDA, cudnn, GPU 설정하기

간만에 다시 설치하려니 이것저것 꼬여서 힘들다.

CUDA 11.0 버전은 왠일인지 다운로드가 안된다. 이건 뭘까...

결국 tensorflow 버전을 2.2.0으로 낮추고 CUDA 10.1 버전을 설치하고 cudnn v7.6.5를 설치하고서야 작동하게 되었다.

 

휴.... 험난한 과정을 거쳐 겨우 GPU 모드에서 학습을 할 수 있게 되었다.

다시 한번 정리하자면, GPU카드가 지원 가능한 CUDA 버전을 확인하는 일 (무조건 최신 버전을 설치하면 안됨)

1. Tensorflow 설치 버전을 확인하기

2. 해당 Tensorflow가 지원하는 CUDA 버전 설치하기

3. cudnn 모듈을 다운받아 설치하기

* 그런데 nvidia-smi 실행시 CUDA Version: 11.2라고 나와 있어서 이걸 설치해야 하는 줄 알았다.

* 또한, insufficient permissions 라는 processes 설명도 뭔가 곤란하게 만듦