Home | 简体中文 | 繁体中文 | 杂文 | Github | 知乎专栏 | Facebook | Linkedin | Youtube | 打赏(Donations) | About
知乎专栏

3.4. Ubuntu

			
cat >> /etc/resolv.conf <<EOF
# Generated by NetworkManager
search netkillerc.cn
nameserver 114.114.114.114
nameserver 8.8.8.8
options timeout:1 single-request-reopen
EOF
			
		
		
root@netkiller:~# apt update -y
root@netkiller:~# apt upgrade -y	
root@netkiller:~# apt autoremove	-y
		
		

3.4.1. Nvidia 驱动

			
root@netkiller:~# ubuntu-drivers devices
ERROR:root:aplay command not found
== /sys/devices/pci0000:00/0000:00:0d.0 ==
modalias : pci:v000010DEd00001EB8sv000010DEsd000012A2bc03sc02i00
vendor   : NVIDIA Corporation
model    : TU104GL [Tesla T4]
driver   : nvidia-driver-418-server - distro non-free
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-470 - distro non-free
driver   : nvidia-driver-545 - distro non-free
driver   : nvidia-driver-535 - distro non-free recommended
driver   : nvidia-driver-570-server - distro non-free
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-450-server - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin			
			
			

			
root@netkiller:~# apt install nvidia-driver-570-server
root@netkiller:~# prime-select nvidia 
			
			
			
sudo prime-select query //查看当前使用显卡
sudo prime-select nvidia //使用nvidia显卡
sudo prime-select intel //使用intel显卡			
			
			
			
root@netkiller:~# lsmod | grep nvidia
root@netkiller:~# modprobe nvidia
root@netkiller:~# lsmod | grep nvidia
nvidia_uvm           1781760  0
nvidia_drm            110592  0
nvidia_modeset       1527808  1 nvidia_drm
nvidia              89739264  2 nvidia_uvm,nvidia_modeset
drm_kms_helper        311296  4 cirrus,nvidia_drm
drm                   622592  5 drm_kms_helper,nvidia,cirrus,nvidia_drm			
			
			

			
apt install nvidia-utils-550		
			
			

			
root@netkiller:~# nvidia-smi
Fri Mar 14 17:50:55 2025       
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 570.86.15              Driver Version: 570.86.15      CUDA Version: 12.8     |
|-----------------------------------------+------------------------+----------------------+
| 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  Tesla T4                       Off |   00000000:00:0D.0 Off |                    0 |
| N/A   65C    P0             31W /   70W |       1MiB /  15360MiB |      8%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
                                                                                         
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+			
			
			

			
root@netkiller:~# /srv/python/bin/python
Python 3.11.0rc1 (main, Aug 12 2022, 10:02:14) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> 			
			
			

			
root@netkiller:~# gpu-manager
last_boot_file: /var/lib/ubuntu-drivers-common/last_gfx_boot
new_boot_file: /var/lib/ubuntu-drivers-common/last_gfx_boot
can't access /opt/amdgpu-pro/bin/amdgpu-pro-px
Looking for nvidia modules in /lib/modules/5.15.0-134-generic/kernel
Looking for nvidia modules in /lib/modules/5.15.0-134-generic/updates/dkms
Found nvidia.ko module in /lib/modules/5.15.0-134-generic/updates/dkms/nvidia.ko
Looking for amdgpu modules in /lib/modules/5.15.0-134-generic/kernel
Looking for amdgpu modules in /lib/modules/5.15.0-134-generic/updates/dkms
Is nvidia loaded? yes
Was nvidia unloaded? no
Is nvidia blacklisted? no
Is intel loaded? no
Is radeon loaded? no
Is radeon blacklisted? no
Is amdgpu loaded? no
Is amdgpu blacklisted? no
Is amdgpu versioned? no
Is amdgpu pro stack? no
Is nouveau loaded? no
Is nouveau blacklisted? yes
Is nvidia kernel module available? yes
Is amdgpu kernel module available? no
Vendor/Device Id: 1013:b8
BusID "PCI:0@0:2:0"
Is boot vga? yes
Vendor/Device Id: 10de:1eb8
BusID "PCI:0@0:13:0"
can't open /sys/bus/pci/devices/0000:00:0d.0/boot_vga
Is boot vga? no
can't open /sys/bus/pci/devices/0000:00:0d.0/boot_vga
Chassis type: "1"
Laptop not detected
Is nvidia runtime pm supported for "0x1eb8"? no
Checking power status in /proc/driver/nvidia/gpus/0000:00:0d.0/power
Runtime D3 status:          ?
Is nvidia runtime pm enabled for "0x1eb8"? no
Takes 0ms to wait for nvidia udev rules completed.
Skipping "/dev/dri/card1", driven by "nvidia-drm"
Skipping "/dev/dri/card0", driven by "cirrus"
Skipping "/dev/dri/card1", driven by "nvidia-drm"
Skipping "/dev/dri/card0", driven by "cirrus"
Skipping "/dev/dri/card1", driven by "nvidia-drm"
Skipping "/dev/dri/card0", driven by "cirrus"
Skipping "/dev/dri/card1", driven by "nvidia-drm"
Skipping "/dev/dri/card0", driven by "cirrus"
Does it require offloading? no
last cards number = 2
Has amd? no
Has intel? no
Has nvidia? yes
How many cards? 2
Has the system changed? No
Unsupported discrete card vendor: 10de
Nothing to do			
			
			

			
root@netkiller:~# apt install nvidia-cuda-toolkit			
			
			

3.4.2. Python 环境

			
cat >> requirements.txt << EOF
ultralytics
onnx
numpy
matplotlib
opencv-python
opencv-python-headless
pillow
httpx
texttable
dominate
EOF
			
			

			
root@netkiller:~# apt install python3.11 python3.11-venv 
root@netkiller:~# python3.11 -m venv /srv/python
root@netkiller:~# /srv/python/bin/pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 -i https://pypi.tuna.tsinghua.edu.cn/simple
root@netkiller:~# /srv/python/bin/pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple	
			
			

			
import torch
torch.cuda.is_available()		
			
			

			
root@netkiller:/opt/原图训练数据# /srv/python/bin/yolo task=detect mode=train model=/srv/medical/model/yolo11m.pt data=/opt/datasets/AllInOne/data.yaml epochs=300 device=0 name=AllInOne project=runs
Ultralytics 8.3.89 🚀 Python-3.11.0rc1 torch-2.6.0+cu124 CUDA:0 (Tesla T4, 14914MiB)