有小伙伴安装cuda时,可能会出现安装失败的情况,这里给大家提供一个思路。
我的安装思路是:安装(查看)驱动 —> 显示驱动对应的CUDA版本
一、检查现有CUDA版本
首先,检查系统中是否已经安装了其他版本的CUDA:
nvcc --version
如果有其他版本的CUDA,你可能需要将其卸载:
sudo apt-get --purge remove "*cublas*" "cuda*" "nsight*" "nvidia*"
二、更新软件包列表
更新软件包列表,以确保你获得最新的软件包信息:
sudo apt-get update
三、安装(查看)驱动
1、查看驱动安装情况
先查看一下自己电脑是否已经安装NVIDIA驱动:
nvidia-smi
如果显示下面这张图片,则表示电脑已安装驱动:
若没显示,则需要先查看自己电脑适合什么版本的驱动并进行下载。
2、下载驱动
1、更新包列表:
sudo apt update
2、安装所需的依赖项:
sudo apt install build-essential dkms
3、添加NVIDIA PPA(可选):
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
4、列出可用的NVIDIA驱动程序:
ubuntu-drivers devices
这将显示适合你系统的NVIDIA驱动程序,如下图:
可挑选适合自己需求的驱动版本进行下载。
5、安装推荐的NVIDIA驱动程序:
假设推荐安装的驱动程序版本是nvidia-driver-535
,你可以使用以下命令进行安装:
sudo apt install nvidia-driver-535
6、重启计算机:
安装完成后,重启计算机以应用驱动程序:
sudo reboot
3、确认是否安装成功
运行:
nvidia-smi
若出现:
则表示安装驱动成功,查看对应版本为CUDA 为12.2版本,去CUDA官网进行下载。
CUDA各版本网址:CUDA Toolkit Archive | NVIDIA Developer
四、安装CUDA
打开网站,点击框选区域(以12.2为例):
按步骤选择对应版本:
下面会弹出对应安装代码,按照代码安装即可:
安装CUDA完成后,检查一下是否安装到路径中:
ls /usr/local/
出现这个,下载完成!