Ubuntu 深度炼丹环境配置

Ubuntu 深度炼丹环境配置

深度炼丹最麻烦的就是环境配置了,下面过程记录了本人进行深度炼丹环境的配置。

首先是安装图形显卡驱动,按以下指令进行便可linux

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install nvidia-396

这样即可以安装完图形显卡驱动。ubuntu

紧接着就是安装CUDA,这是一个比较麻烦的工做。首先是从官网上下载最新的CUDA版本,下面的命令也能够,可是安装获得的版本一般不是最新的。bash

sudo apt install nvidia-cuda-toolkit

下载runfile便可,安装过程较为简单。为了验证安装是否成功,能够按以下步骤进行测试:测试

查看cuda 的版本号,输入以下指令便可:code

nvcc -V

编译CUDA samples。找到你安装CUDA的过程当中,选择的samples保存位置,进入后运行make命令,而后能够在Samples里面找到bin/x86_64/linux/release/目录,并切换到该目录  运行deviceQuery程序,sudo ./deviceQuery pass即经过。get

最后一步最烦人,目前亲自试过的比较正确的方式是以下it

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
相关文章
相关标签/搜索