개발환경에 맞게 설치하자.
[CUDA 11.8 , cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x ]
[CUDA 12.4 , cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x ]
모든버전의 cuDNN : https://developer.nvidia.com/rdp/cudnn-archive
# CUDA 11.8
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.1 -c pytorch -c nvidia
# CUDA 12.4
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia
'Dev > Deep Learning' 카테고리의 다른 글
Pytorch 스터디 (0) | 2025.01.03 |
---|---|
컴퓨터 비전 프로젝트를 위한 8개의 오픈소스 의료 이미지 라이브러리 (1) | 2024.12.13 |
전처리 툴 v 0.1 (0) | 2024.11.07 |
Anomalib (1) | 2024.08.18 |
이상 탐지 관련 오픈소스 프로젝트들 (0) | 2024.08.18 |
댓글