본문 바로가기
Dev/Deep Learning

CUDA cuDNN 개발환경

by wenect 2024. 12. 22.

개발환경에 맞게 설치하자.


[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

 

댓글