pytorch - Torch.cuda.is_available()가 False..?

아직 괜찮은 pytorch 로 연산을 많이하는 프로그램을 만들진 않았지만!

구색을 갖추기 위해서 CUDA를 사용하려고 한다.

심지어 더 좋은 GPU가 달린 노트북으로 변경까지 했다.

 

pytorch를 설치하고

아래 명령어를 입력해보자.

python -m torch.utils.collect_env

뭐어?!

is CUDA available이 False라고?

 

이건 같이 설치되는 CUDA 11.8버전이면 이런 현상이 발생한다고 한다.

그래서 설치한 패키지를 그대로 uninstall 하고 아래처럼 CUDA 11.7로 다시 설치했다.

 

다시 pytorch 의 환경변수 수집 명령어를 쳐보면

CUDA available이 True로 출력되는 것을 확인할 수 있다!

물론 GPU가 CUDA를 지원해야 가능하다.

 

특정 라이브러리에 CUDA버전이 강요되는게 많아서 슬픔..

CUDA 관련해서는 범용성이 많이 떨어지므로 버전확인 필요