Ubuntu18.04安装安装cuda和cudnn

时间:2021-06-24 18:33:54   收藏:0   阅读:0

一.安装cuda11.0

cuda下载
此处的cuda下载请根据自己的系统信息下载,否则使用会有问题。
技术图片

安装说明 :

    wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
    sudo sh cuda_11.0.3_450.51.06_linux.run

1.启动安装程序,一直按空格到最后,输入accept接受条款
技术图片

2.不安装nvidia图像驱动,之前已经安装过了
技术图片

3.安装完成后会有安装的日志记录输出, cat /var/log/cuda-installer.log,查看无报错一般就是安装成功了,
技术图片

4.配置环境变量

vim ~/.bashrc

加入路径:

    export PATH=/usr/local/cuda/bin/:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64/

刷新配置:source ~/.bashrc

5.cuda验证

    nvcc -V

技术图片

5.测试

首先安装g++

    apt-get install g++

    #编译并测试设备 deviceQuery:
    cd /usr/local/cuda-11.0/samples/1_Utilities/deviceQuery
    sudo make
    ./deviceQuery
     
    #编译并测试带宽 bandwidthTest:
    cd ../bandwidthTest
    sudo make
    ./bandwidthTest

如果这两个测试的最后结果都是Result = PASS,说明CUDA安装成功啦

二.安装cuDNN

下载cudnn
技术图片

查找nvcc -V对应的版本下载
技术图片

1.安装

解压到cuda文件夹内 运行以下命令,进行安装

    cd cuda/include/ 
    sudo cp ./include/cudnn.h /usr/local/cuda/include #执行即可,不需要做更改
    sudo cp ./lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

2.建立软连接

可以到下/usr/local/cuda/lib64/看一下自己的文件

    cd /usr/local/cuda/lib64/ 
    sudo rm -rf libcudnn.so libcudnn.so.8
    sudo ln -s libcudnn.so.8.0.5 libcudnn.so.8
    sudo ln -s libcudnn.so.8 libcudnn.so

保存后,创建链接文件

    sudo vim /etc/ld.so.conf.d/cuda.conf 

按a进入插入模式,增加下面一行


    /usr/local/cuda/lib64 

按esc退出插入模式,按:wq保存退出

最后在终端输入


    sudo ldconfig 

后续Anaconda3等的安装请参考此处:https://www.cnblogs.com/KdeS/p/14741567.html

文章参考地址:https://blog.csdn.net/pursuit_zhangyu/article/details/79362128

评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!