我们在跑模型的时候,需要用GPU并行加速,需要用显卡跑,这就需要安装显卡驱动。

参考方法,视频:https://www.bilibili.com/video/BV16Y411M7SC/?spm_id_from=333.788

显卡驱动便捷的方法

  打开软件和更新,选择附加驱动,就能够把当前设备所能够使用的驱动搜出来。

  如果没有选择,安装的是开源版本,但是建议安装470版本,更加稳定,是配型很好,18.04和20.04都是可以用的

  选择这个版本之后,点击应用更改。

  判断显卡驱动安装好了,调出终端,输入nvidia-smi,会显示当前驱动的信息

显卡驱动安装的第二个方法

  去官网下载安装包

  怎么运行这个文件

    1、查看文件权限  ll ./NVIDIA-Linux-x86_64-520.56.06.run

    2、赋予执行权限 sudo chmod +x ./NVIDIA-Linux-x86_64-520.56.06.run

    3、执行文件 sudo ./sudo ./NVIDIA-Linux-x86_64-520.56.06.run

    4、判断显卡驱动安装好了,输入nvidia-smi,会显示当前驱动的信息

安装cuda toolkit archive

  wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.ru

  sudo sh cuda_11.8.0_520.61.05_linux.run

  运行run文件按照上文

补充:cuda和cudnn都可以在conda里面安装,只安装显卡驱动也没什么问题。如果是跑c++程序的话就需要安装了,c++好像没有虚拟环境这一说,这个我不太清楚。

 

posted on 2022-10-25 14:29  鱼干儿兄  阅读(302)  评论(0编辑  收藏  举报