[python] torch.cuda.is_available() 이 false일 때 해결법

본인이 cuda + cudnn을 설치했고 pytorch 까지 설치했는데, torch.cuda.is_available() 를 실행했을 때 false를 반환한다면 이 글을 참고하면 좋겠다.

 

 

원인은 다운 받은 torch버전이 본인의 cuda버전을 지원하지 않거나, cpu전용 이기 때문이다.

(버전 이슈 or gpu지원x이다.)

 

 

해결 방법

예를 들어 torch 버전이 2.1.1이고, cuda버전이 12.1인데 false가 뜬다고 하자.

설치되어 있는 torch 버전을 보면 2.1.1 or 2.1.1+cpu 이렇게 되어있을거다.

버전을 2.1.1+cu121로 바꿔주면 해결가능

 

 

본인의 cuda버전을 확인하고, 본인의 cuda버전이 적혀있는 버전의 torch를 아래 링크의 명령어를 사용해서 설치하자.

 

(사실 본인 cuda 버전이 11.6이다 그럼 12.1버전의 cuda를 지원하는 torch를 설치해도 괜찮다. 즉, 본인 cuda보다 높거나 같은 버전의 cuda를 지원하는 torch를 설치해도 문제없더라)

 

 

ex) cuda버전이 12.1이고 torch==2.1.0버전이 필요한 경우

 

pip install torch==2.1.0 대신에 아래의 명령어를 실행하자.

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121

 

 

설치가 끝나고 torch 버전을 보면 torch 2.1.1+cu121이렇게 되어있을거다.

이제 torch.cuda.is_available()를 실행하면 true를 반환한다!

 

 

본인이 필요한 torch 버전에 맞는 명령어를 찾고 싶으면 아래 링크를 누르면 나온다.

 

 

 

최신 버전 설치 명령어

https://pytorch.org/

PyTorch

 

이전 버전 설치 명령

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

Previous PyTorch Versions

 

 

 

최신 버전 (누르면 나옴)

 

이전 버전 (누르면 나옴)