yuns

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

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를 위하여 :를 사용한다.

반응형

'linux, github, module 에러 정리' 카테고리의 다른 글

Jupyter Lab 설치하기  (0) 2021.04.28
linux pid의 사용자 확인하기  (0) 2021.04.21
TensorRT 설치하기  (0) 2021.04.20
github프로젝트에서 install하기  (0) 2021.04.18
Tokenizers 설치/에러/해결  (0) 2021.04.18
Comments