CUDA pytorch

Python 버전과 CUDA 버전이 요청한 PyTorch 버전과 호환되지 않을 때

1. 파이썬 버전 확인

python --version

Python 3.8.19 

2. cuda 버전 확인

nvcc --version

cuda 11.2

3. CUDA 11.2와 호환되는 PyTorch 버전이 없어서, 가장 가까운 버전(CUDA 11.3) 설치

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113

 

4. PyTorch가 제대로 설치되었는지 확인