# 切换root账号操作
# 进行一些依赖库的升级也很必要

# 卸载驱动和cuda
nvidia-uninstall
cuda-uninstaller

# 关闭 nouveau
lsmod | grep nouveau
vim /lib/modprobe.d/dist-blacklist.conf
# 去掉 nvidiafb 注释
# 添加:
blacklist nouveau
options nouveau modeset=0

# 重建initramfs image:
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)

# 文本模式
systemctl set-default multi-user.target # 切换回图形界面:systemctl set-default graphical.target
reboot

#安装
chmod a+x NVIDIA-Linux-x86_64-470.86.run
./NVIDIA-Linux-x86_64-470.86.run

# nvidia 安装成功
nvidia-smi

# 安装cuda:
sh cuda_11.5.1_495.29.05_linux.run #默认更新driver从470到495

# 成功之后按照提示添加路径:
vim /etc/profile
export CUDA_HOME=/usr/local/cuda-11.5
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
source /etc/profile

# 安装cudnn(别用rpm安装!!!老老实实压缩包解压):
tar -zxvf cudnn-11.5-linux-x64-v8.3.0.98.tgz

# 将解压后中的部分文件复制到目标路径下,并更改权限
cp cuda/include/cudnn*.h /usr/local/cuda/include 
cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 
chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

# 检查:
nvidia-smi
nvcc -V

# 大功告成!
  • nvidia-smi
    请添加图片描述

  • nvcc -V
    请添加图片描述

  • tensorflow检验
    请添加图片描述

  • torch检验:
    请添加图片描述

Logo

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

更多推荐