☰

是否有一个 PPA 可以将 nvidia-cuda-toolkit 与最新的 cuda (11.1) 一起用于 Ubuntu 20.04 (Focal)？

tag-icon • tag-icon nvidia cuda
是否有一个 PPA 可以将 nvidia-cuda-toolkit 与最新的 cuda (11.1) 一起用于 Ubuntu 20.04 (Focal)？

官方软件包似乎不是最新的（https://packages.ubuntu.com/search?keywords=nvidia-cuda-toolkit) 但在使用较新的硬件时，使用最新的 cuda 似乎很重要。

Nvidia 正在提供软件包，但不能保证它们能运行良好或能与 Ubuntu 的 nvidia 驱动程序管理集成。

是否有尝试在 PPA 中创建一个 ubuntu 包，这样我们就不必手动安装并单独解决冲突？（包会处理这个问题）

答案1

  1. 打开终端并删除系统上可能存在的任何 NVIDIA 痕迹。

    sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt-get remove --autoremove nvidia-*

  2. 在您的系统上设置正确的 CUDA PPA。

    export DISTRO=ubuntu2004
export ARCHITECTURE=x86_64
sudo apt update
sudo add-apt-repository ppa:graphics-drivers
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCHITECTURE}/7fa2af80.pub 
sudo bash -c "echo deb\ http://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCHITECTURE}/\ / > /etc/apt/sources.list.d/cuda.list"  
sudo bash -c "echo deb\ http://developer.download.nvidia.com/compute/machine-learning/repos/${DISTRO}/${ARCHITECTURE}\ / > /etc/apt/sources.list.d/cuda_learn.list"

  3. 安装 CUDA 11.1 包。

    sudo apt update
sudo apt -y install cuda-11-1
sudo apt install libcudnn8

  4. 在 '.profile' 文件中指定 CUDA 的 PATH。通过运行打开该文件sudo nano ~/.profile并在文件末尾添加以下几行：

    # set PATH for cuda 11.1 installation
if [ -d "/usr/local/cuda-11.1/bin/" ]; then
    export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

    使用键盘组合键Ctrl+O然后按 将Enter文件保存到当前位置。使用键盘组合键Ctrl+X退出 nano。

  5. 重新启动sudo reboot并检查安装的版本。

    CUDA：

    nvcc  --version

    NVIDIA 驱动程序：

    nvidia-smi

    libcudnn：

    /sbin/ldconfig -N -v $(sed ‘s/:/ /’ <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn

答案2

感谢 karel 的回答。在第 2 步中，我遇到了 GPG 错误：“公钥不可用”。

我关注了 nvidia 问题中 jtran1999 的评论1632正确添加密钥和 CUDA PPA。也许这会帮助其他遇到同样错误的人：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update

然后继续执行步骤 3，一切正常！！

相关内容