메뉴 바로가기 본문 바로가기

KAIST 전산학부

Site Location

고객지원 게시판

게시판

KCLOUD2 GPU Driver 및 CUDA 설치 (2022. 02. 28. 기준)

작성자
administrator
작성일
2022-02-28 09:21
조회
319

KCLOUD2의 GPU 인스턴스의 GPU Driver(ver 495.29.05) 및 CUDA(ver 11.5.1), CUDNN(8.3) 설치 방법을 안내해 드립니다.

해당 명령어를 순서대로 실행하시거나, 스크립트 화 시켜서 한번에 실행하시면 됩니다.

RTX3090, TITAN X, TITAN RTX 에서 모두 해당 방법으로 설치 가능합니다.

 


wget http://172.10.4.9:8080/v1/AUTH_93538585e16c43ab822104c6890f3d7c/cudnn/cuda_11.5.1_495.29.05_linux.run
wget http://172.10.4.9:8080/v1/AUTH_93538585e16c43ab822104c6890f3d7c/cudnn/cudnn-linux-x86_64-8.3.1.22_cuda11.5-archive.tar.xz

apt update
apt install -y build-essential

chmod +x cuda_11.5.1_495.29.05_linux.run
./cuda_11.5.1_495.29.05_linux.run --silent --driver --toolkit

echo -e "\n## CUDA and cuDNN paths" >> ~/.bashrc
echo 'export PATH=/usr/local/cuda-11.5/bin:${PATH}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.5/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc

source ~/.bashrc

tar -xvf cudnn-linux-x86_64-8.3.1.22_cuda11.5-archive.tar.xz

cp -P ./cudnn-linux-x86_64-8.3.1.22_cuda11.5-archive/include/cudnn*.h /usr/local/cuda-11.5/include
cp -P ./cudnn-linux-x86_64-8.3.1.22_cuda11.5-archive/lib/libcudnn* /usr/local/cuda-11.5/lib64/
chmod a+r /usr/local/cuda-11.5/include/cudnn*.h
chmod a+r /usr/local/cuda-11.5/lib64/libcudnn*

첨부파일 :
전체 0