호환 CUDA ver. 등 확인

https://en.wikipedia.org/wiki/CUDA

https://developer.nvidia.com/cuda-toolkit-archive

https://docs.nvidia.com/deploy/cuda-compatibility/index.html#binary-compatibility__table-toolkit-driver

Windows

NVIDIA 설정

NVIDIA Driver 설치:

• Windows 10 with RTX 3060
  • NVIDIA 560.70 드라이버 다운로드.
• Windows 11 with RTX 3050 Ti
  • NVIDIA 561.09 드라이버 다운로드.

CUDA Toolkit 설치:

• CUDA 11.8 설치.
• 설치 후 시스템 환경 변수에 CUDA 경로(CUDA_PATH) 추가 확인

cuDNN 설치:

• cuDNN 8.9.7 다운로드 후 CUDA 폴더 내 include, lib, bin에 복사.

1. Conda 환경 설정:
   • conda create -n bori python=3.10

• PyTorch 설치:

  (conda 더 안정적) conda 환경에서는 Python 3.10 ≤ 권장

  python=3.10

  windows, linux

  conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia osx conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 -c pytorch

  • pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

Python 3.11 환경이 꼭 필요하다면

pip로 설치하는 것이 더 안전