主页 / ubuntu / 问题 / 1292120
Accepted
lgautier
Asked: 2020-11-14 12:49:14 +0800 CST

是否有用于 Ubuntu 20.04 (Focal) 的具有最新 cuda (11.1) 的 nvidia-cuda-toolkit 的 PPA？

  • 772

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

Nvidia 正在提供软件包，但不能保证它们会很好地工作或将与 Ubuntu 的 nvidia 驱动程序管理集成。

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

nvidia cuda

2 个回答

  1. Best Answer

    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 的路径。通过运行打开文件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-smi

      libcudnn:

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

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

    我按照 nvidia issue 1632中 jtran1999 的评论正确添加了密钥和 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 步，一切正常！！

    • 0

相关问题