0
点赞
收藏
分享

微信扫一扫

ubuntu安装CUDA

1.安装 gcc、 g++编译器

apt-get install gcc

apt-get install g++

apt-get install make

2.确认是否安装软件包

dpkg -l gcc

dpkg -l g++

dpkg -l make

3.检查 nouveau 模块是否加载, 已加载则先禁用

lsmod | grep nouveau

4.禁用系统自带的 nouveau 模块

vi /etc/modprobe.d/blacklist.conf

在文本最后添加以下内容:

blacklist nouveau

options nouveau modeset=0

保存退出, 执行以下命令生效:

update-initramfs -u

5.重启操作系统

init 0

6.安装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.1/local_installers/cuda-repo-ubuntu1804-11-6-local_11.6.1-510.47.03-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1804-11-6-local_11.6.1-510.47.03-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

7.测试 cuda 安装是否正确

nvcc -V

思腾合力(SITONHOLY)是AI服务器与HPC基础架构解决方案商,专注人工智能服务器领域,拥有自主品牌AI服务器及通用X86服务器,公司为NVIDIA产品官方授权经销商,主营深思IW系列GPU服务器产品可支持1~20颗GPU,适用于深度学习训练及推理等场景,覆盖服务器、静音工作站等多种产品形态,能够满足客户全场景需求。公司扎根高性能计算领域多年,已经打造出了一套完全自主的软硬件结合的产品生态。

举报

相关推荐

0 条评论