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
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)