LabHub

블로그

Install Nvidia driver on ubuntu22.04

한국어English日本語

RTX4070 MaxQ Ubuntu 22.04

lspci | grep -i nvidia
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get install linux-headers-$(uname -r)
sudo apt install --only-upgrade ubuntu-drivers-common
sudo ubuntu-drivers autoinstall
sudo apt install nvidia-cuda-toolkit

재부팅

$ nvidia-smi
Fri Nov 29 03:07:20 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.120                Driver Version: 550.120        CUDA Version: 12.4     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 4070 ...    Off |   00000000:01:00.0 Off |                  N/A |
| N/A   48C    P8              3W /  115W |       9MiB /   8188MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A      1354      G   /usr/lib/xorg/Xorg                              4MiB |
+-----------------------------------------------------------------------------------------+

docker 설치


sudo apt install docker.io
sudo apt install docker-buildx
sudo chmod 666 /var/run/docker.sock

nvidia-docker2 설치

## GPG 키와 저장소 추가!
sudo apt install curl
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list


# 마지막으로 다시 업데이트!
sudo apt-get update


# nvidia-docker2 설치
sudo apt-get install -y nvidia-docker2

테스트용 도커 명령어

## Test!

docker run --rm --gpus all ubuntu:18.04 nvidia-smi

댓글

아직 댓글이 없습니다.

로그인하면 댓글을 쓸 수 있습니다