登录  /  注册
博主信息
博文 291
粉丝 0
评论 0
访问量 431774
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
Ubuntu下安装CUDA
原创
2195人浏览过

镜像下载、域名解析、时间同步请点击 阿里云开源镜像站

注意:
Ubuntu18.04默认GCC-7,由于CUDA 9.x不支持GCC-7(下载页面没有对ubuntu18的支持),所以需要安装低版本的5或者<= 6.3.0,并设置为默认版本。否则应安装10.0以上版本CUDA

1.查看是否安装了cuda

  1. # 法1
  2. cat /usr/local/cuda/version.txt
  3. # 法2
  4. nvcc --version

2.若没有安装,则查看是否有N卡驱动,若无N卡驱动,则到软件与更新 -> 附加驱动中安装驱动

3.查看N卡驱动支持的cuda版本

  1. nvidia-smi

如下图支持最高CUDA版本是11.4

file

4.到cuda-toolkit-archive,选择需要的CUDA下载

5.如下图,选择runfile(local),并使用生成的指令进行下载和安装

file

若第1步提示Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.,选择continue,在下一步中去除driver项,之后选择install:

file

安装完成后,显示如下:

file

6.在~/.bashrc文件中添加如下环境变量:

  1. export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
  2. export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

7.验证是否安装成功

  1. nvcc -V

file

8.安装cuDNN,到cudnn-archive下载和CUDA对应的版本

  1. 1.Navigate to your <cudnnpath> directory containing the cuDNN tar file.
  2. 2.Unzip the cuDNN package.
  3. $ tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz
  4. 3. Copy the following files into the CUDA toolkit directory.
  5. $ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
  6. $ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
  7. $ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

按照↑教程,可下载cuDNN Library for Linux (x86_64)用复制的方式安装,使用如下命令查看安装版本

  1. cat /usr/local/cuda-11.3/include/cudnn.h | grep CUDNN_MAJOR -A 2

9.也可下载Runtime Library和Developer Library的deb包,使用sudo dpkg -i xxx.deb命令进行安装,使用如下命令查看安装版本

  1. cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

10.多版本CUDA安装重复上述步骤即可,最后通过更改第6步中环境变量切换CUDA版本

本文转自:https://blog.csdn.net/CC977/article/details/122789394

本博文版权归博主所有,转载请注明地址!如有侵权、违法,请联系admin@php.cn举报处理!
全部评论 文明上网理性发言,请遵守新闻评论服务协议
0条评论
作者最新博文
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号

  • 登录PHP中文网,和优秀的人一起学习!
    全站2000+教程免费学