重庆企业网站开发服务wordpress文章tags
先检查显卡驱动是否安装
nvidia-smi
- 如上显示就代表驱动已安装完成，同时为下一步安装cuda确认信息，查看右上角CUDA Version的版本，我目前是12.8，就代表我目前可以安装的cuda版本不能高于12.8。
- 如果没有如上显示，就代表驱动未安装，可执行以下命令使用apt安装
添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update查看可安装的驱动版本
ubuntu-drivers devices
建议直接使用recommended推荐版本
sudo apt install nvidia-driver-XXX之后再次输入nvidia-smi，若还是看不到，可尝试重启
安装CUDA
CUDA Toolkit 12.9 Downloads | NVIDIA Developer
首先是确定版本，上述链接为nvidia安装最新cuda的网址。
按照之前说的，确认自己需要的版本，之前的版本可以从Archive of Previous CUDA Releases中找。
进入之后按照自己系统的配置选择选项
查看ubuntu版本可用以下命令
lsb_release -a安装完成后还需要配置环境变量
sudo nano ~/.bashrc在末尾添加如下两句,注意其中的xx需要更换为自己cuda的版本号
export PATH=/usr/local/cuda-xx.x/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xx.x/lib64:$LD_LIBRARY_PATH
退出时使用ctrl+x,询问是否保存修改，输入y，再问写入文件名称，确认后直接回车
回到命令行，输入如下命令，让修改立即生效
source ~/.bashrc最后输入下述命令，验证cuda的安装
nvcc -V
安装cuDNN
cuDNN 9.10.1 Downloads | NVIDIA Developer
安装cudnn类似cuda，直接进入官网按照命令下载即可，最后一步对照自己的cuda执行
如需验证，可先安装torch包，然后执行下述代码
import torch
print(torch.backends.cudnn.version()) # This will print the cuDNN version if available