我们在跑模型的时候,需要用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++好像没有虚拟环境这一说,这个我不太清楚。