在 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 Toolkit 套件以及驅動程式套件。
cuda-編譯器-11-6 安裝所有 CUDA 編譯器套件。
cuda-libraries-11-6 安裝所有運行時 CUDA 庫包。
cuda-libraries-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

相關內容