ubuntu gpu 환경 셋업 시 check 명령어

Lee Hyun Joon ·2022년 11월 11일
0

https://data-science-hi.tistory.com/202

위 사이트 참고를 통해 ubuntu 셋업 진행했습니다..

처음부터 알려주시는대로 진행하시면 문제 없이 진행 (셋업 버전 cuda-11.8 & cudnn 8.5.0 & toolkit)

cuda 버전 설치 후 check 명령어

$ nvidia-smi   # nvidia 명령어 
$ cat /proc/driver/nvidia/version

cuda toolkit 설치 후 check 명령어

$ nvcc -V # nvcc --version 

cudnn 설치 후 check 명령어

$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # later version용 
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 # 이전 명령어 

cudnn check 명령어 내용 출처
https://stackoverflow.com/questions/31326015/how-to-verify-cudnn-installation

profile
우당탕탕 개발 지망생

0개의 댓글