[Ubuntu - Pytorch] 파이토치 cuda와 cuDNN이 설치 되어있는데도 GPU가 안잡힐때 / torch.cuda.is_available() Fasle 해결
·
OS/Ubuntu
일단 기본적으로 파이토치를 설치하고 잘 연결되어있는지 확인하기 위해 해당 코드를 실행한다 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_..