CUDA: CUDNN安装测试

CUDNN安装测试


细节
  • 环境

    说明备注
    CUDAToolkit11.2.r11.2A40 vGPU/CentOS-8.x
    cudnn-local-repo-rhel8-8.5.0.96-1.0-1.x86_64.rpmCUDNN8.5-CUDA11.7官网下载
    rpm -Uvh安装/var/cudnn-local-repo-rhel8-8.5.0.96获得三个rpm: libcudnn8-8.5.0.96-1.cuda11.7.x86_64.rpm,libcudnn8-devel-8.5.0.96-1.cuda11.7.x86_64.rpm,libcudnn8-samples-8.5.0.96-1.cuda11.7.x86_64.rpm
  • 安装步骤

    
    #安装解压repo包
    rpm -Uvh cudnn-local-repo-rhel8-8.5.0.96-1.0-1.x86_64.rpm
    cd /var/cudnn-local-repo-rhel8-8.5.0.96
    mkdir tmp && cd tmp
    
    #将rpm解压到本地路径
    rpm2cpio ../libcudnn8-8.5.0.96-1.cuda11.7.x86_64.rpm | cpio -div
    rpm2cpio ../libcudnn8-devel-8.5.0.96-1.cuda11.7.x86_64.rpm | cpio -div
    rpm2cpio ../libcudnn8-samples-8.5.0.96-1.cuda11.7.x86_64.rpm | cpio -div 
    
    
    #将库(.so,.a)和软连接拷贝到CUDAToolkit对应目录下
    cp -d ./usr/lib64/* /usr/local/cuda/lib64/.
    
    #将头文件(.h)和软连接拷贝到CUDAToolkit对应目录下
    cp -d ./usr/include/* /usr/local/cuda/include/.
    
    ##-------------------------------------------
    cd /usr/local/cuda/lib64
    
    ##建立软连接: 旧版本兼容
    ln -s libcudnn.so.8 libcudnn.so
    
    ##---------------------------------------------
    
    cd /usr/local/cuda/include 
    
    
    ##建立软连接: 旧版本兼容
    ln -s cudnn_v8.h cudnn.h
    ln -s cudnn_version_v8.h cudnn_version.h
    ln -s cudnn_ops_train_v8.h cudnn_ops_train.h
    ln -s cudnn_ops_infer_v8.h cudnn_ops_infer.h
    ln -s cudnn_cnn_train_v8.h cudnn_cnn_train.h
    ln -s cudnn_cnn_infer_v8.h cudnn_cnn_infer.h
    ln -s cudnn_backend_v8.h cudnn_backend.h
    ln -s cudnn_adv_train_v8.h cudnn_adv_train.h
    ln -s cudnn_adv_infer_v8.h cudnn_adv_infer.h
    
    ##--------------------------------------------
    
    ##测试
    cd /var/cudnn-local-repo-rhel8-8.5.0.96/tmp/usr/src/cudnn_samples_v8/conv_sample
    make 
    ./conv_sample
    
    
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值