linux, github, module 에러 정리

CUDA, CUDNN 설치하기 및 로컬(.bashrc)에 저장

yuuuun 2021. 4. 20. 14:07
반응형

1) CUDA설치하기

2) CUDNN설치하기

https://developer.nvidia.com/rdp/cudnn-archive에 들어가서 로그인 이후 설치하기

아래 코드를 이용하여  tar파일 풀기

tar -zxvf cudnn[각 버전에 따라 완성되는 폴더명이 달라짐]

이후에는 cudnn의 파일을 cuda 파일에 넣어주어야 함

cp lib64/* /usr/local/cuda/lib64
cp include/* /usr/local/cuda/include

cuda와 cudnn 설치 완성

.bashrc에 저장하는 방법

export CUDA_DIR="/usr/local/[cuda directory]"
export PATH=$CUDA_DIR"/bin":$PATH
export LD_LIBRARY_PATH=$CUDA_DIR"/lib64":$LD_LIBRARY_PATH

데이터의 concat를 위하여 :를 사용한다.

반응형