首页 > 系统相关> 文章详细

Ubuntu环境下深度学习环境配置不完全指北

2020-05-03 22:58:39 阅读：370 来源： 互联网

标签：指北 sudo Ubuntu 环境 cuDNN cuda NVIDIA local 下载

1.NVIDIA 驱动下载[可选]

在NVIDIA驱动下载页面选择并下载与自己显卡匹配的驱动

NVIDIA驱动下载页面

2.CUDA Toolkit 下载

在NVIDIA Developer官网找到CUDA Toolkit下载页面
选择如下配置即可获得CUDA Toolkit下载地址

CUDA Toolkit选项

3.cuDNN 下载

cuDNN需要登录NVIDIA Developer账号才能下载，登陆后打开cuDNN下载地址

cuDNN下载页面

同意cuDNN Software License Agreement后选择cuDNN Library for Linux版本下载

4.NCCL 下载[可选]

NCCL同样需要登录NVIDIA Developer账号才能下载，登陆后打开NCCL下载地址

NCCL下载页面

同意Software License Agreement后选择O/S agnostic local installer版本下载

至此已获得如下文件

NVIDIA-Linux-x86_64-440.31.run
cuda_10.1.243_418.87.00_linux.run
cudnn-10.1-linux-x64-v7.6.5.32.tgz
nccl_2.4.8-1+cuda10.1_x86_64.txz[可选]

下面我们进行安装

5.NVIDIA 驱动安装

1.自动方式

执行如下命令

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-drivers devices
sudo apt-get install nvidia-driver-xxx

2.手动方式

sudo bash NVIDIA-Linux-x86_64-440.31.run

6.CUDA Toolkit 安装

sudo bash cuda_10.1.243_418.87.00_linux.run --override

安装完成后执行如下命令添加PATH和LD_LIBRARY_PATH

sudo nano ~/.bashrc

在尾部添加

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

最后执行source 使环境变量生效

source ~/.bashrc

7.cuDNN 安装

tar xvzf cudnn-10.1-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/* /usr/local/cuda/include

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

8.NCCL 安装

tar Jxvf nccl_2.4.8-1+cuda10.1_x86_64.txz

cd nccl_2.4.8-1+cuda10.1_x86_64/

标签：指北,sudo,Ubuntu,环境,cuDNN,cuda,NVIDIA,local,下载
来源： https://www.cnblogs.com/smallsung/p/12824370.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9