一、安装cuda

1.先查看有没有安装CUDA 在终端输入命令

nvcc -V 

2、在官网下安装cuda

CUDA Toolkit Archive | NVIDIA Developer

 选择想要安装的版本,进去之后依次根据自己的配置选择

注意最后一项选择deb(local),如果选择runfile(local)可能会出错

 根据给出的命令,在桌面右键打开终端输入

 3、修改环境变量

vim ~/.bashrc

点击 i 进行输入

export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

点击esc退出编辑,再输入

:wq

保存退出,输入

source ~/.bashrc

4、检查是否安装成功

重启电脑,然后查看nvidia

nvidia-smi

查看cuda安装

nvcc -V

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐