深度学习图像分类开源项目(1)-深度学习环境搭建(cuda10 + cudnn + anaconda3)

作者信息
广东科学技术职业学院 计算机工程技术学院(人工智能学院)AI应用实验室

主要从事深度学习算法、计算机视觉、智能算法应用与部署,基于人工智能,计算机视觉算法web前后端应用开发,android开发,微信小程序开发,可以承接相关的项目开发业务。QQ群:971601256

成员:
刘炳辉,主要研究方向:机器视觉。

指导老师:胡建华,余君

转载:https://blog.csdn.net/weixin_43483316/article/details/103595514?utm_source=app

深度学习环境搭建(cuda10 + cudnn + anaconda3)

建议下载好需要用到的显卡驱动、cuda、cudnn、anaconda3等安装包文件,放在用户的目录下,记住该目录路径。本文的显卡是RTX 2080 super、cudnn7+cuda10.1

N卡显卡驱动下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn
cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载地址(需要账号):https://developer.nvidia.com/cudnn
anconda3下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

1. 安装显卡驱动(这里以2080 super显卡为例)

  1. 删除原有驱动 (这里假设有安装驱动)

打开终端(按住Ctrl + Alt + T)

a) 禁用nouveau (安装NVIDIA需要把系统自带的驱动禁用)
打开文件: sudo vim /etc/modprobe.d/blacklist.conf
文本末尾追加:

blacklist nouveau
options nouveau modeset=0

然后保存退出(:wq),打开终端:lsudo update-initramfs -u
重启Ubuntu系统:reboot

b) 验证nouveau是否已禁用(不禁用nouveau安装显卡会报错)
打开终端输入:lsmod | grep nouveau

在这里插入图片描述
如果没有任何显示,则表示禁用正常
如果有显示驱动,则回去重复前面的步骤,重新禁用显卡

  1. 进入命令行界面,按Ctrl + Alt + F1

关闭图形界面: sudo service lightdm stop

给驱动文件加权限: sudo chmod a+x ./驱动文件名

执行驱动文件: sudo ./驱动文件名 -no-opengl-files

(安装过程中,所有选择使用默认选项)

  1. 挂载显卡: modprobe nvidia

查看显卡信息: nvidia-smi

在这里插入图片描述
如果显示上图信息,则成功安装显卡驱动,没有则安装失败。
安装失败的话,不要继续下面的步骤,重复前面卸载显卡和安装显卡的步骤

  1. 开启图形界面:sudo service lighdm start

驱动安装参考:

https://blog.csdn.net/xunan003/article/details/81665835

安装并配置CUDA

给cuda文件加权限 sudo chmod a+x cuda文件名

安装cuda sudo sh cuda文件名

安装过程中的选项
在这里插入图片描述
2. 测试CUDA

我用的是cuda10.0,所以我的安装路径是:/usr/local/cuda-10.0

cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

  1. 出现如下信息,则安装成功

在这里插入图片描述

  1. 安装完成后,设置环境变量

  2. 打开终端,输入: sudo vim ~/.bashrc

然后在文档最后面加入下面几句命令:(路径为自己安装的cuda目录)

如我的目录为:usr/local/cuda-10.0/,则输入:

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

然后保存退出,在终端运行:source ~/.bashrc

  1. 检查是否配置成功:新开一个终端,输入nvcc --version,如果显示下面的文子就说明安装成功了
    在这里插入图片描述

下面是官方给的在线下载,安装cuda10.1的方法,仅供参考

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

安装cudnn加速库(这里以cudnn7 + cuda10.1为例)

  1. 解压安装cudnn sudo dpkg -i cudnn文件名

​//后面两个步骤是最后验证安装时需要的(不一定只有这一种方式验证)
sudo dpkg -i libcudnn7-dev_7.6.4.38-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.4.38-1+cuda10.1_amd64.deb
验证安装

sudo cp -r /usr/src/cudnn_samples_v7/ /home/wdong/
cd /home/wdong/mnistCUDNN
sudo make
./mnistCUDNN

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值