ubuntu 安装cuda cudnn

要在Ubuntu上安装CUDA和cuDNN，可以按照以下步骤进行操作：

1. 首先，根据您的显卡型号选择合适的NVIDIA驱动程序。您可以在终端中运行以下命令来获取有关本机显卡及其驱动程序的相关信息：

   ubuntu-drivers devices

根据输出的信息选择推荐的驱动程序版本，然后运行以下命令自动安装驱动程序：

   sudo ubuntu-drivers autoinstall

或者，如果您想安装特定版本的驱动程序，可以运行以下命令：

   sudo apt install nvidia-driver-<version>

请将`<version>`替换为您想要安装的驱动程序版本号。

2. 安装CUDA。您可以从NVIDIA官方网站下载适用于Ubuntu的CUDA安装程序。选择适合您的系统的版本，并使用以下命令进行安装：

   wget <CUDA installer URL>
   sudo sh <CUDA installer filename>

请将`<CUDA installer URL>`替换为CUDA安装程序的下载链接，将`<CUDA installer filename>`替换为下载的CUDA安装程序的文件名。

3. 安装cuDNN。您可以从NVIDIA开发者网站下载适用于Ubuntu的cuDNN库。选择适合您的CUDA版本的cuDNN版本，并使用以下命令进行安装：

   sudo cp cuda/include/cudnn*.h /usr/local/cuda-<CUDA version>/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<CUDA version>/lib64
   sudo chmod a+r /usr/local/cuda-<CUDA version>/include/cudnn.h

请将`<CUDA version>`替换为您安装的CUDA版本号。

安装完成后，您可以使用以下命令来验证CUDA和cuDNN的安装是否成功：

nvidia-smi

如果显示了有关显卡和驱动程序的信息，则表示安装成功。

请注意，安装CUDA和cuDNN可能需要一些额外的配置步骤，具体取决于您的系统和需求。建议您参考NVIDIA官方文档和安装指南以获取更详细的信息和指导。