centos 7.6 / ubuntu 18 安装cuda 11.4

发布时间:2024年01月22日
1、下载cuda驱动 cuda_11.4.0_470.42.01_linux.run
   cuda下载地址:https://developer.nvidia.com/cuda-downloads?target_os=Linux
 
2、禁用nouveau内核模块
   vim /etc/modprobe.d/blacklist-nouveau.conf
   # 1、添加如下内容:
   blacklist nouveau
   options nouveau modeset=0
    
   update-initramfs -u   # ubuntu 18 需执行该命令,否则无法禁掉该内核模块
  
   # 2、重启虚机
   reboot
   # 验证nouveau禁用成功
   lsmod |grep nouveau
   # 没有显示:禁用成功
 
3、安装cuda
   chmod +x cuda_11.4.0_470.42.01_linux.run
   sh cuda_11.4.0_470.42.01_linux.run
 
4、配置环境变量
   参考链接:https://blog.csdn.net/FYH666520/article/details/127701315
   vim ~/.bashrc
   增加下面两行内容,并保存
   export PATH=/usr/local/cuda-11.4/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
   使配置文件生效
   source ~/.bashrc
   重启电脑
   reboot
 
5、输入nvidia-smi 验证是否安装成功

文章来源:https://blog.csdn.net/hanweixiao/article/details/135742917
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。