Linux torch 버전 확인하는 법

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을 보여줍니다.