在 Ubuntu 18.04 中强制使用 CUDA 工具包版本 11.6,最新的版本是 9.1

在 Ubuntu 18.04 中强制使用 CUDA 工具包版本 11.6,最新的版本是 9.1

我刚刚按照 nvidia 网站上的说明安装了 cuda:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

然后我安装了 nvidia-cuda-toolkit:

sudo apt install nvidia-cuda-toolkit

所有这些都没有问题,但它安装了该工具包的 9.1 版本:

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

我需要该工具包的 11.1 或更高版本,但是似乎 9.1 是存储库中的最新版本:

$ sudo apt install nvidia-cuda-toolkit
Reading package lists... Done
Building dependency tree       
Reading state information... Done
nvidia-cuda-toolkit is already the newest version (9.1.85-3ubuntu1).

我如何强制安装最新版本?

这是我的系统的详细信息:

$ cat /etc/*release
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=18.04
DISTRIB_CODENAME=bionic
DISTRIB_DESCRIPTION="Ubuntu 18.04.6 LTS"
NAME="Ubuntu"
VERSION="18.04.6 LTS (Bionic Beaver)"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 18.04.6 LTS"
VERSION_ID="18.04"
HOME_URL="https://www.ubuntu.com/"
SUPPORT_URL="https://help.ubuntu.com/"
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
VERSION_CODENAME=bionic
UBUNTU_CODENAME=bionic

$ uname -m
x86_64

$ uname -r
4.15.0-167-generic

提前致谢!

答案1

如果你正在关注官方指南,您部分遵循了它。

您必须安装以下元包之一:

元包 目的
库达 安装所有 CUDA 工具包和驱动程序包。处理发布后升级到下一版本的 cuda 包。
CUDA-11-6 安装所有 CUDA 工具包和驱动程序包。在安装其他版本的 CUDA 之前,版本将保持为 11.6。
cuda-工具包-11-6 安装开发 CUDA 应用程序所需的所有 CUDA 工具包。不包括驱动程序。
cuda-工具-11-6 安装所有 CUDA 命令行和可视化工具。
cuda-运行时-11-6 安装运行 CUDA 应用程序所需的所有 CUDA 工具包以及驱动程序包。
cuda-编译器-11-6 安装所有 CUDA 编译器包。
cuda-库-11-6 安装所有运行时 CUDA 库包。
cuda-库-dev-11-6 安装所有开发 CUDA 库包。
cuda 驱动程序 安装所有驱动程序包。处理驱动程序包发布后升级到下一版本的事宜。

最简单的方法是

sudo apt-get install cuda

以完全安装最新的 CUDA 11.6 版本。例如,其编译器将安装在以下位置:

$ which nvcc 
/usr/local/cuda-11.6/bin/nvcc

需要 11.6 版本

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Fri_Dec_17_18:16:03_PST_2021
Cuda compilation tools, release 11.6, V11.6.55
Build cuda_11.6.r11.6/compiler.30794723_0

你还应该知道nvidia-cuda-toolkit包裹来自Ubuntu 存储库如下所示

$ apt-cache policy nvidia-cuda-toolkit 
nvidia-cuda-toolkit:
 Installed: (none)
 Candidate: 9.1.85-3ubuntu1
 Version table:
    9.1.85-3ubuntu1 500
       500 http://archive.ubuntu.com/ubuntu bionic/multiverse amd64 Packages

所以你不应该使用它。

答案2

如果您设置环境变量 -PATHLD_LIBRARY_PATH,这将会起作用。

例子:

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

相关内容