헬창 개발자
nvidia-smi, nvcc --version cuda 버전이 다른 경우 본문
1. nvidia-smi
2. nvcc --version
CUDA는 2개의 API를 가지고 있는데, 하나는 runtime API이고 다른 하나는 driver API이다.
각 API가 각자의 version을 가지고 있다.
- nvidia-smi 의 CUDA는 runtime API를 나타내며, GPU 드라이버에 의해 설치된다.
- nvcc의 CUDA는 CUDA toolkit에 의해 설치된다.
설치한 CUDA 버전을 확인하는건 2. nvcc --version 명령어가 옳다.
* 만약 nvcc --version 명령어가 안될 경우 CUDA toolkit 을 설치해 줘야 한다.
호환되는 버전은 아래 링크 참고
Wikiwand - CUDA
In computing, CUDA is a proprietary parallel computing platform and application programming interface (API) that allows software to use certain types of graphics processing units (GPUs) for accelerated general-purpose processing, an approach called general
www.wikiwand.com
CUDA - Wikipedia
From Wikipedia, the free encyclopedia Parallel computing platform and programming model In computing, CUDA (originally Compute Unified Device Architecture) is a proprietary[1] parallel computing platform and application programming interface (API) that all
en.wikipedia.org
이제 올바른 cuda버전을 알았으니 해당 링크에 들어가 설치명령어를 입력해주면 된다.
Previous PyTorch Versions
Installing previous versions of PyTorch
pytorch.org
올바르게 설치가 됐다면 파이썬에 다음 코드를 작성해보면
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
라고 뜬다.
'공부방' 카테고리의 다른 글
| 허깅페이스 자주 사용하는 파라미터 (0) | 2024.08.01 |
|---|---|
| LangChain 문서 분할 종류 (0) | 2024.07.29 |
| 키워드 검색(Keyword Search)과 의미 검색(Semantic Search)의 차이 (0) | 2024.06.20 |
| LCEL (LangChain Expression Language) (0) | 2024.06.18 |
| LLM fine-tunning (2) | 2024.06.14 |
