import torch
print(torch.version.cuda) # PyTorch가 사용하는 CUDA 버전
print(torch.backends.cudnn.version()) # PyTorch가 사용하는 cuDNN 버전
print(torch.cuda.is_available()) # GPU 사용 가능 여부
print(torch.cuda.device_count()) # 사용 가능한 GPU 개수
print(torch.cuda.get_device_name(0)) # GPU 모델명 확인
위 명령어를 쳐주시면 됩니다.
중요!!!ㅁ
현재 사용하고 있는 CUDA 버전이 아닌 PyTorch가 권장하는 버전이 나오게 됩니다.
CUDA 버전은 아래의 명령어로 확인하실 수 있습니다.
nvcc -V
아래의 명령어는 그래픽카드 현재 사용을 볼 수 있습니다!
nvidia-smi

여기서 보이는 CUDA Version은 현재 사용 중인 CUDA의 Version이 아닌 제가 사용하고 있는 그래픽 카드가 권장하는 CUDA의 Version을 보여줍니다.
'Tools & Setup > Setting' 카테고리의 다른 글
| Dataset 수집 (0) | 2025.04.02 |
|---|---|
| Vibe Coding을 위한 Cursor 소개 및 설치 방법 (0) | 2025.03.27 |
| Ubuntu(우분투)에서 새 계정 생성하는 방법 (0) | 2025.03.07 |
| Ubuntu TensorFlow 버전에 호환되는 cuDNN 설치 (0) | 2025.03.06 |
| Ubuntu에서 Pytorch(파이토치) 설치하기 (6) | 2025.03.06 |