ubantu20.04，cuda12.2 docker镜像安装cuda， cudann

在 Ubuntu 20.04 的 Docker 镜像中安装 CUDA 12.2 和 cuDNN 可以通过使用 NVIDIA 提供的官方镜像来实现。虽然 NVIDIA 官方仓库可能尚未直接提供包含 CUDA 12.2 的镜像，但可以通过手动下载并安装 CUDA Toolkit 和 cuDNN 来完成配置。 ### 准备工作 确保系统已安装以下组件： - **NVIDIA 驱动**：支持 CUDA 12.2 的版本（如 535 或更高）。 - **Docker Engine**：按照 [Install Docker Engine on Ubuntu](https://docs.docker.com/engine/install/ubuntu/) 指南进行安装[^2]。 - **NVIDIA Container Toolkit**：用于启用容器内 GPU 支持。 ```bash 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 sudo apt-get install -y nvidia-docker2 sudo systemctl restart docker ``` ### 获取基础镜像并创建容器 尽管官方提供的镜像可能未包含 CUDA 12.2，可以基于较新的 CUDA 版本镜像（如 `nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04`）作为起点，并在其上手动安装 CUDA 12.2 工具包和 cuDNN。 ```bash sudo docker pull nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04 sudo docker run -it --gpus all --name cuda_env -e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04 /bin/bash ``` ### 在容器内安装 CUDA 12.2 进入容器后，下载并安装 CUDA 12.2 Toolkit 和 cuDNN。 1. 下载 CUDA 12.2 Toolkit： ```bash 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 ./cuda_12.2.0_535.54.03_linux.run --silent --toolkit ``` 2. 设置环境变量（可添加到 `.bashrc` 中）： ```bash export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 3. 安装 cuDNN： 从 [NVIDIA cuDNN 页面](https://developer.nvidia.com/cudnn) 下载适用于 CUDA 12.2 的 cuDNN 包，并将其复制到容器中进行安装。 ```bash tar -xzvf cudnn-linux-x86_64-8.x.x.x_cuda12-archive.tar.gz sudo cp cudnn-linux-x86_64-8.x.x.x_cuda12-archive/include/cudnn*.h /usr/local/cuda-12.2/include sudo cp cudnn-linux-x86_64-8.x.x.x_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.2/lib64 sudo chmod a+r /usr/local/cuda-12.2/include/cudnn*.h /usr/local/cuda-12.2/lib64/libcudnn* ``` ### 验证安装 运行以下命令验证 CUDA 和 cuDNN 是否正确安装： ```bash nvcc --version cat /usr/local/cuda-12.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 ``` 输出应显示 CUDA 12.2 和 cuDNN 的版本信息。 ### 总结 通过上述步骤，可以在 Ubuntu 20.04 的 Docker 环境中成功安装 CUDA 12.2 和 cuDNN。此过程涉及使用 NVIDIA 提供的基础镜像、手动下载并安装 CUDA Toolkit 和 cuDNN，并配置相应的环境变量以确保其正常运行。 ---