当前位置:   article > 正文

linux下安装CUDA和cudnn_linux安装cuda和cudnn

linux安装cuda和cudnn

linux下安装CUDA和cudnn

2、安装CUDA10.1
在CUDA官网选择系统对应的版本下载CUDA
我的系统是Ubuntu18.04 64位,我选择下载最新版本的CUDA10.1:
在这里插入图片描述
输入以下命令安装CUDA
sudo sh cuda_10.1.105_418.39_linux.run
显卡驱动已在上面安装了,这里一定要选择不安装驱动,如下:
在这里插入图片描述
安装完后会有警告提示信息,忽略就行。是因为在安装CUDA时没有选择安装显卡驱动。
安装完后,在.bashrc文件末尾添加环境变量
sudo gedit ~/.bashrc

export PATH=“/usr/local/cuda-10.1/bin: P A T H " e x p o r t L D L I B R A R Y P A T H = " / u s r / l c o a l / c u d a − 10.1 / l i b 64 : PATH" export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.1/lib64: PATH"exportLDLIBRARYPATH="/usr/lcoal/cuda10.1/lib64:LD_LIBRARY_PATH”

保存退出后,输入以下命名
source ~/.bashrc
输入以下命令,测试是否安装成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

如下图 Result = PASS表示CUDA安装成功。
在这里插入图片描述

3、安装cuDNN7.5
在官网下载cuDNN,选择自己对应的系统和已安装的CUDA版本下载cuDNN。(要注册才能下载)

我选择的是cuDNN v7.5.1 (April 22, 2019), for CUDA 10.1 for Linux
在这里插入图片描述
下载完成后,输入以下命令解压文件

tar -zxvf cudnn-10.1-linux-x64-v7.5.1.10.tgz

输入以下命名,将cudnn相关文件拷贝到已安装的cuda文件中

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ #解压后的文件夹名为cuda

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h #增加所有用户对文件的可执行权限

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/397964
推荐阅读
相关标签
  

闽ICP备14008679号