프로그래밍/파이썬 공부

CUDA 버전 변경법

이자다 2024. 2. 7. 00:00
반응형

https://tw0226.tistory.com/79

 

[CUDA] 윈도우 환경에서 CUDA 버전 변경

어떤 github의 딥러닝 코드는 tensorflow 1.14 버전을 썼는데, 1.14 버전은 CUDA 11버전이 호환되지 않아서 1. python 버전 변경 2. tensorflow 재설치 3. CUDA 버전 번경 (11 -> 10) 의 절차가 필요하였다. 나의 상황

tw0226.tistory.com

 

위 링크에 자세히 나와 있다.

 

간략히 정리하자면 

 

1. 원하는 버전의 CUDA를 설치한다.

 

2. 고급 시스템 설정 -> 환경변수에서 CUDA_PATH를 이용하고자 하는 버전으로 변경하고, Path 내에서도 사용하고 싶은 버전을 상위로 올린다.

 

3. 설정이 완료되면 cmd 창으로 nvcc -V 명령어로 cuda 버전이 몇으로 나오는지 확인한다. PyTorch로 print(torch.cuda.version)를 사용해도 확인할 수 있다고 한다.

반응형