3090 ubuntu20.04 cuda

要在 Ubuntu 20.04 上使用 NVIDIA GeForce RTX 3090 GPU 进行深度学习开发，需要安装最新版本的 NVIDIA 驱动和 CUDA 工具包。以下是安装步骤：

1. 安装 NVIDIA 驱动程序

从 NVIDIA 官网下载最新版本的驱动程序，并按照指示进行安装。可以使用以下命令检查所需的驱动程序版本：

   $ ubuntu-drivers devices

然后安装所需的驱动程序版本（如使用 nvidia-driver-460）：

   $ sudo apt install nvidia-driver-460

2. 安装 CUDA 工具包

从 NVIDIA 官网下载所需版本的 CUDA 工具包，并按照指示进行安装。例如，您可以下载 CUDA Toolkit 11.1.1 for Ubuntu 20.04 x86_64。

具体安装步骤如下：

   $ sudo dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.1-455.32.00-1_amd64.deb
   $ sudo apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
   $ sudo apt-get update
   $ sudo apt-get install cuda

3. 配置环境变量

添加以下行到`~/.bashrc`文件中：

   export PATH=/usr/local/cuda/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后使用以下命令更新环境变量：

   $ source ~/.bashrc

4. 验证安装

运行以下命令来验证 CUDA 工具包和 NVIDIA 驱动程序的安装：

   $ nvcc -V
   $ nvidia-smi

如果显示了 CUDA 工具包和驱动程序的版本信息，则表示安装成功。