카테고리 없음

CUDA (nvcc, nvidia-smi) install

자월현 2022. 3. 2.

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

이걸 확인하는 과정이 있지만, 대부분의 경우 이걸 만족한다고 볼 수 있다. 하지만 짚고 넘어가고 싶다면 확인해보자.

lspci | grep -i nvidia
uname -m && cat /etc/*release

위의 코드로 nvidia 가능한 GPU인지, OS는 맞는지를 확인한다. 대부분 맞을 것이다.

gcc --version

그다음 gcc가 설치되어있는지 확인한다. 보통 한번 밀었다가 다시 깔면 gcc가 없다. 없는 경우 밑의 코드를 활용하여 설치하자.

sudo apt install gcc

nvidia cuda toolkit 다운로드해서 설치하기.

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=deb_network

이거를 참고하면 두개 드라이버?의 차이를 좀더 알 수 있지 않을까.. 희망해본다.

이쯤하고 nvidia-smi 를 하면 정보가 뜬다. 만약 안뜨면 reboot하면 된다.

nvidia-smi

그러고서 /usr/local에 cuda가 잘 설치되어 있다면, vim ~/.bashrc 한 후 파일에 밑의 두 줄을 붙여넣고서 저장하고,
(i 를 누르면 insert 가능, shift+ctrl+v로 붙여넣기, esc 누른 후 :wq 누르면 저장)
source ~/.bashrc로 효력발행?을 한다.

vim ~/.bashrc
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

그러면 nvcc -V로 쿠다가 잘 설치된걸 확인할 수 있다.

nvcc -V

보통 cuda 버전 다른것이 필요할 때, toolkit만 설치하면 된다.

 

댓글