[python] 현재 PC에 설치되어 있는 CUDA 버전 확인하기

여러 버전의 CUDA를 설치하고 지우고를 반복하다보니 제 노트북에 어떤 버전의 CUDA가 설치되어 있는지 간혹 헷갈리더라고요. 그래서 현재 설치되어 있는 CUDA 버전을 확인하는 방법을 소개해드리려고 합니다. 

 

cmd로 명령 프롬프트를 여시고, nvcc -version을 타이핑하시면 됩니다. 그러면 다음과 같이 버전을 확인할 수 있습니다. 

 

 

10.0 버전의 CUDA가 설치되어 있었네요. ㅎㅎ tensorflow나 pytorch를 gpu 상에서 작동하게 하려면 정확한 CUDA 버전을 알아둘 필요가 있습니다. 

 

 

<참고자료>

[1] https://www.it-swarm.dev/ko/cuda/%EC%BF%A0%EB%8B%A4-%EB%B2%84%EC%A0%84%EC%9D%84-%EA%B5%AC%ED%95%98%EB%8A%94-%EB%B0%A9%EB%B2%95/941876836/, it-swarm.dev, "쿠다 버전을 구하는 방법?"