NVIDIA CUDA工具包提供了开发环境,可供创建经 GPU 加速的高性能应用。借助 CUDA 工具包,您可以在经 GPU 加速的嵌入式系统、台式工作站、企业数据中心、基于云的平台和 HPC 超级计算机中开发、优化和部署应用。此工具包中包含多个 GPU 加速库、多种调试和优化工具、一个 C/C++ 编译器以及一个用于在主要架构(包括 x86、Arm 和 POWER)上构建和部署应用的运行时库。借助多 GPU 配置中用于分布式计算的多项内置功能,科学家和研究人员能够开发出可从单个 GPU 工作站扩展到配置数千个 GPU 的云端设施的应用。本博文示例安装在Ubuntu18.04系统下安装cuda11.4,官网查看安装要求如下:
root@testuat:/opt# uname -m && cat /etc/*release
x86_64
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=18.04
DISTRIB_CODENAME=bionic
DISTRIB_DESCRIPTION=“Ubuntu 18.04.5 LTS”
NAME=“Ubuntu”
VERSION=“18.04.5 LTS (Bionic Beaver)”
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME=“Ubuntu 18.04.5 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
root@testuat:/opt# gcc --version
gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
Copyright © 2017 Free Software Foundation, Inc.
This is free software; see the source for copying conditions. There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
root@testuat:/opt# sudo apt-get install linux-headers-$(uname -r)
正在读取软件包列表… 完成
正在分析软件包的依赖关系树
正在读取状态信息… 完成
linux-headers-5.4.0-81-generic 已经是最新版 (5.4.0-81.91~18.04.1)。
…
根据系统环境官网选择英伟达cuda工具包
root@testuat:/opt# wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
root@testuat:/opt# sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
正在选中未选择的软件包 cuda-repo-ubuntu1804-11-4-local。
(正在读取数据库 … 系统当前共安装有 214569 个文件和目录。)
正准备解包 cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb …
正在解包 cuda-repo-ubuntu1804-11-4-local (11.4.1-470.57.02-1) …
正在设置 cuda-repo-ubuntu1804-11-4-local (11.4.1-470.57.02-1) …
root@testuat:/opt# sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
OK
root@testuat:/opt# sudo apt-get update
当前最新版本为11.4.1
root@testuat:/opt# sudo apt-get -y install cuda
…
update-initramfs: Generating /boot/initrd.img-5.4.0-81-generic
正在处理用于 libc-bin (2.27-3ubuntu1.4) 的触发器 …
正在处理用于 man-db (2.8.3-2ubuntu0.1) 的触发器 …
nvidia-smi命令可以安装cuda之前执行,可以看到显卡驱动对应支持的cuda版本为11.4,如果安装其他版本nvidia-smi命令执行会报错mismatch。
root@vanfonuat:/opt# nvidia-smi
(base) vanfon@testuat:~$ source ~/.bashrc
(base) vanfon@testuat:~$ nvcc -V
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2021 NVIDIA Corporation
Built on Wed_Jul_14_19:41:19_PDT_2021
Cuda compilation tools, release 11.4, V11.4.100
Build cuda_11.4.r11.4/compiler.30188945_0
root@testuat:/usr/local/cuda# cd /etc/ld.so.conf.d/
root@testuat:/etc/ld.so.conf.d# vim cuda.conf
#添加如下内容
/usr/local/cuda/lib64
root@testuat:/etc/ld.so.conf.d# sudo ldconfig
在步骤5的链接中下拉到最后,点击Archive of Previous CUDA Releases,进入历史版本列表。
wuhs@jqxxpc:/usr/local$ vim ~/.bashrc
#添加cuda环境变量
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
(base) bdsc@jqxxpc:/usr/local$ source ~/.bashrc
wuhs@jqxxpc:~$ nvcc -V
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2022 NVIDIA Corporation
Built on Tue_Mar__8_18:18:20_PST_2022
Cuda compilation tools, release 11.6, V11.6.124
Build cuda_11.6.r11.6/compiler.31057947_0
如果您发现该资源为电子书等存在侵权的资源或对该资源描述不正确等,可点击“私信”按钮向作者进行反馈;如作者无回复可进行平台仲裁,我们会在第一时间进行处理!
添加我为好友,拉您入交流群!
请使用微信扫一扫!