ubuntu 安装cuda 与 cuDNN

0.   准备工作

      禁用BIOS中的secure boot,因为此方法使用第三方源安装显卡驱动,不禁止secure boot会导致安装的驱动不能使用,禁用也不会有多大安全隐患。

       禁用nouveau,这是ubuntu默认使用的开源显卡驱动,和nvidia驱动一起使用可能导致黑屏,所以禁掉。

sudo gedit /etc/modprobe.d/blacklist.conf

在文本的最后一行添加

blacklist nouveau
options nouveau modeset=0

执行命令使禁用生效并且重启:

$ sudo update-initramfs -u
$ sudo reboot

 查看是否生效

$ lsmod | grep nouveau

没有输出,禁用成功

2. 安装Nvidia驱动

查看本机支持的Nvidia驱动

ubuntu-drivers devices

我的显示推荐460,安装CUDA10.2,依次执行下述命令安装460驱动:

sudo apt-get purge nvidia* 
sudo add-apt-repository ppa:graphics-drivers/ppa 
sudo apt-get update 
sudo apt-get install nvidia-driver-460

然后重启

sudo shutdown -r now

3.安装CUDA10.2

        首先删除系统可能剩余的cuda 

sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

    进入nvidia官网:CUDA Toolkit 12.4 Downloads | NVIDIA Developer ,这里默认进去是最新的版本,需要选择适合自己的CUDA,同时需要注册,有时候能进去,有时候进不去。进不去的等一会再试试。

选择后会有相应的命令, 分别执行以下命令

重启后执行

$ nvidia-smi #如果出现GPU列表,则驱动安装成功

4.安装cuDNN

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值