龙空技术网

Ubuntu下 CUDA 安装

冷冻工厂 126

前言:

今天你们对“ubuntulncuda软链接”大概比较讲究,小伙伴们都想要学习一些“ubuntulncuda软链接”的相关资讯。那么小编同时在网络上搜集了一些有关“ubuntulncuda软链接””的相关知识,希望同学们能喜欢,你们一起来了解一下吧!

导读

本文将介绍如何在Ubuntu下安装深度学习需要使用的CUDAcudnn

1. CUDA安装前往官网[1]下载符合自己系统版本的CUDA,根据自己电脑配置和系统情况,选择对应的选项,如下图:

CUDA

根据网页的安装提示操作,如下图:

安装提示

打开终端进入到CUDA安装文件所在文件夹,赋予其可执行权利

sudo  chmod +x cuda_10.0.130_410.48_linux.run
安装
sudo ./cuda_10.0.130_410.48_linux.run
修改环境变量
sudo vi ~/.bashrcexport CUDA_HOME=/usr/local/cuda-10.0 # 根据版本修改export LD_LIBRARY_PATH=${CUDA_HOME}/lib64export PATH=${CUDA_HOME}/bin:${PATH}# 更新生效source ~/.bashrc
查看安装的版本信息
nvcc -V# cat /usr/local/cuda/version.txt

nvcc信息

2. cudnn安装官网[2]下载与CUDA对应的cudnn版本(要注册账号才能下载)打开终端进入cudnn下载目录,对其进行解压

tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8mv cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8 cudnn
将解压后的文件复制到cuda目录下
sudo cp cudnn/include/cudnn*.h /usr/local/cuda/include sudo cp -P cudnn/lib/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
查看版本 (可能不显示,但不影响使用)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
参考资料

[1]CUDA官网:

[2]cudnn:

标签: #ubuntulncuda软链接