[Ubuntu - Pytorch] 파이토치 cuda와 cuDNN이 설치 되어있는데도 GPU가 안잡힐때 / torch.cuda.is_available() Fasle 해결

일단 기본적으로 파이토치를 설치하고 잘 연결되어있는지 확인하기 위해 해당 코드를 실행한다

import torch


print("Torch version:{}".format(torch.__version__))
print("cuda version: {}".format(torch.version.cuda))
print("cudnn version:{}".format(torch.backends.cudnn.version()))
print(torch.cuda.is_available())

나는 처음에 해당 코드를 실행시 쿠다와 cuDNN 을 설치했음에도 안잡혔었다..

 

-우분투 18.04 기준으로

 

CUDA 버전 확인

nvcc -V

cuDNN 버전 확인

cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

CUDA와 cuDNN 이 정상적으로 설치 되어 있는 경우라면 파이토치 버전이 잘못된것임..

파이토치 버전 뒤에  +cu116 과 같이 쿠다 버전이 적혀야한다

 

그래서 다운그레이드 해줌

 

해결방법

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

https://pytorch.org/get-started/previous-versions/

PyTorch

해당사이트에서 자신의 쿠다 버전과 OS 환경에 맞게 설치해주면 된다.

 

최종적으로 이렇게 정상적이게 출력되면 끝!!