[Error] torch.cuda.is_available() return False（Try multiple versions）

 

import torch
torch.cuda.is_available()

 

했을 때 False가 뜨는 경우는 pytorch에서 cuda가 제대로 사용할 수있는 상태라고 볼 수 있습니다.

 

pytorch 또는 cuda 설치를 다시 살펴봐야 하고, GPU 모델 번호, 그래픽 드라이버 버전, cuda 버전 및 pytorch 버전이 모두 일치해야 합니다. 

 

Ubuntu 18.04 환경에서는

sudo apt-get install -y datacenter-gpu-manager
sudo systemctl --now enable nvidia-dcgm
sudo systemctl status nvidia-dcgm

sudo apt-get install cuda-drivers-fabricmanager
sudo systemctl start nvidia-fabricmanager
sudo systemctl status nvidia-fabricmanager

 

으로 해결했다고 하는데, 될 때도 있고 안 될 때도 있는 것 같습니다.

 

가장 간단한 방법은 다시 재설치해주는 것입니다.

 

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

 

지긋지긋한 쿠다 버전!