Python으로 PyTorch, Tensorflow, Python, CUDA, cudnn 버전 확인

PyTorch 버전 확인

import torch
print(torch.__version__)

 

 

Tensorflow 버전 확인

import tensorflow as tf
print(tf.__version__)

 

 

Python 버전 확인

import sys
print(sys.version)

 

 

Python 경로 확인

import sys
print(sys.executable)

 

 

CUDA & Driver 버전 확인

1. nvidia-smi

- Driver version, CUDA version을 확인할 수 있다.

 

 

 

2. nvcc

- CUDA version을 확인할 수 있다.

 

 

 

3. PyTorch를 이용해 확인

import torch
print(torch.version.cuda)

 

 

cudnn 버전 확인

import torch
print(torch.backends.cudnn.version())

 

 

관련된 글

nvidia-smi와 nvcc로 본 CUDA version이 다를 때

nvidia-smi와 nvcc로 본 CUDA version이 다를 때

 

CUDA, driver, PyTorch + Tensorflow 호환되는 version 찾고 설치(업그레이드, 다운그레이드)하기

CUDA, driver, PyTorch + Tensorflow 호환되는 version 찾고 설치(업그레이드, 다운그레이드)하기