赞
踩
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/cuda−10.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*
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。