ubuntu24.04安装nvidia cuda和cudnn
全新安装的ubuntu 24.04,最近安装完stable-diffusion后,在使用命令./webui.sh启动时报错:
1 | RuntimeError: Torch is not able to use GPU |
经过查询得知,stable-diffusion-webui默认你已经安装好了cuda,而在ubuntu下安装完nvidia的私有驱动后,并不会安装cuda和cudnn库。
使用nvidia-smi查询,显示的cuda版本只是最高能兼容的版本,并不代表已经安装好cuda了。
由此也就有了本文。
设置nvidia官方源
使用nvidia官方安装源的方式,来在线安装nvidia 驱动、cuda和cudnn。
