一、安装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

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐