카테고리 없음
CUDA (nvcc, nvidia-smi) install
목차
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

댓글