跳转到内容

Ubuntu/Nvidia

来自UbuntuStudioCN
oss>Brucekomike2025年3月2日 (日) 22:45的版本
(差异) ←上一版本 | 最后版本 (差异) | 下一版本→ (差异)

ubuntu repo

driver

export nv_version=550
sudo apt install  nvidia-driver-$nv_version nvidia-utils-$nv_version

cuda

nvidia-cuda-toolkit

nvidia repo

instlalation

sudo apt-key del 7fa2af80
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt install coda-toolkit
sudo apt-get install zlib1g
sudo apt-get -y install cudnn9-cuda-12

bashrc

export PATH=/usr/local/cuda-12.8/bin${PATH:+:${PATH}}

driver

opencv-python-cuda

https://gist.github.com/minhhieutruong0705/8f0ec70c400420e0007c15c98510f133
https://gist.github.com/raulqf/f42c718a658cddc16f9df07ecc627be7
https://github.com/cudawarped/opencv-python-cuda-wheels
export pkg_install="sudo apt-get install -y"
$pkg_install ffmpeg
$pkg_install libavcodec-dev libavformat-dev libavutil-dev libswscale-dev libavresample-dev
$pkg_install python3-dev python3-skbuild

$pkg_install build-essential cmake pkg-config unzip yasm git checkinstall  # generic tools
$pkg_install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev  # required
$pkg_install python3-dev python3-numpy python3-pip
$pkg_install python3-testresources
$pkg_install libjpeg-dev libpng-dev libtiff-dev
$pkg_install libavcodec-dev libavformat-dev libswscale-dev libavresample-dev
$pkg_install libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
$pkg_install libxvidcore-dev x264 libx264-dev libfaac-dev libmp3lame-dev libtheora-dev 
$pkg_install libfaac-dev libmp3lame-dev libvorbis-dev
$pkg_install libopencore-amrnb-dev libopencore-amrwb-dev
proxy.sh git clone --recursive https://github.com/cudawarped/opencv-python-cuda-wheels.git --depth 1
cd opencv-python-cuda-wheels
python3 -m venv --system-site-packages .venv
source .venv/bin/activate
export "CMAKE_ARGS=-DWITH_CUDA=ON -DCUDA_ARCH_BIN=5.0;5.2;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0;10.0;12.0 -DCUDA_ARCH_PTX=12.0"
export ENABLE_CONTRIB=1
export ENABLE_ROLLING=1
python setup.py bdist_wheel --py-limited-api=cp37