栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 系统运维 > 运维 > Linux

windows / linux 安装 cuda/cudnn

Linux 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

windows / linux 安装 cuda/cudnn

cuda/cudnn安装
  • ubuntu18.04安装cuda/cudnn
    • 安装cuda
    • 安装cudnn

ubuntu18.04安装cuda/cudnn 安装cuda
  • 进入官方下载页面选择需要的cuda版本,cuda下载,注意可能需要登录英伟达账号。
  • 选择好需要的cuda信息后,会生成安装命令:例如我需要的版本是cuda11.3.1,在命令行终端运行下面的第一个命令即可下载,下载完成后运行第二个命令进行安装。
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run
  • 安装流程及注意事项:
  1. 安装前注意禁用 nouveau,验证方法:lsmod | grep nouveau 无输出说明已禁用。禁用方法如下图:

  2. 执行安装脚本进行安装:sudo sh cuda_11.3.1_465.19.01_linux.run

  3. 先accept,安装项里不要选择安装驱动,其他都安装。

  4. 设置环境变量:在~/.bashrc文件添加如下路径,修改后注意输入:sudo source ~/.bashrc生效

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64
export PATH=$PATH:/usr/local/cuda-11.3/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.3
  • 检测是否安装成功:如果Result = PASS则安装成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery 
安装cudnn
  • 进入官方下载地址,cudnn下载,根据cuda版本选择合适的cudnn版本进行下载,例如我选择for CUDA 11.x,cuDNN lib… for linux(x86_64)
  • 下载后解压缩:tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
  • 然后进行拷贝操作:主要将解压后对应的文件拷贝到之前安装的cuda里面
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
  • 测试安装是否成功:查看cudnn版本 ,如果出现版本信息表示成功。
 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
  • 至此cuda和cudnn安装完毕!
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/361913.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号