ubuntu22.04 安装 docker cuda
系统初始化
sudo apt update
sudo apt install vim git proxychains openssh-server curl build-essential python3 -y
安装cuda12.2.0
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
chmod +x cuda_12.2.0_535.54.03_linux.run
sudo sh cuda_12.2.0_535.54.03_linux.run
gedit ~/.bashrc
export PATH = $PATH :/usr/local/cuda-12.2/bin
export LD_LIBRARY_PATH = $LD_LIBRARY_PATH :/usr/local/cuda-12.2/lib64
安装cudnn
nvcc -V
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.hh /usr/local/cuda/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
安装miniconda3
./Miniconda3-py310_24.1.2-0-Linux-x86_64.sh
gedit ~/.condarc
source ~/.bashrc
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
conda clean -i
安装docker
sudo dpkg -i containerd.io_1.6.31-1_amd64.deb
sudo dpkg -i docker-ce-cli_26.1.0-1~ubuntu.20.04~focal_amd64.deb
sudo dpkg -i docker-ce_26.1.0-1~ubuntu.20.04~focal_amd64.deb
sudo dpkg -i docker-compose-plugin_2.26.1-1~ubuntu.20.04~focal_amd64.deb
sudo dpkg -i docker-buildx-plugin_0.14.0-1~ubuntu.20.04~focal_amd64.deb
sudo mkdir -p /etc/docker
sudo tee /etc/docker/daemon.json <<- 'EOF'
{
"registry-mirrors": ["https://xxxx.mirror.aliyuncs.com"]
}
EOF
sudo systemctl daemon-reload
sudo systemctl restart docker
安装docker-cuda库
sudo curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker
测试docker-cuda是否正常安装
sudo docker run --rm --gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi
下载yolo最新的库
sudo docker pull ultralytics/ultralytics
sudo docker run -it --ipc = host --gpus all ultralytics/ultralytics