1.安装SSH

为了通过ssh工具远程连接服务器

sudo apt-get install openssh-server

 2.vim wget

ubuntu自带的是vi,可能会在编辑时不能移动光标。

wget可用于下载东西:wget -c url

sudo apt-get install vim

vim --version

apt-get update

apt-get install wget

 3.安装显卡驱动

删除ubuntu缺省的驱动。

sudo apt-get purge nvidia*

添加ubuntu的ppa源用于下载驱动

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

 搜索最近的驱动,其中有推荐驱动

ubuntu-drivers devices

 可以选择自动安装驱动或者选择自己所需要的版本

sudo ubuntu-drivers autoinstall   或者

sudo apt-get install nvidia-版本号

进行重启之后查看是否有驱动信息

sudo reboot 

nvidia-smi

4.安装cuda

官网中搜索相应版本,会有对应的installation instruction.

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

进行安装选择中,第一个驱动不要安装,因为之前已经安装了。后面几个全部都装。

 

 设置环境变量:

a.对于多个服务器用户使用公用dl环境的情况,在/etc/profile中添加cuda的路径;

b.对于自己单独使用,只需要在.bashrc中添加cuda路径就可以。

a.sudo vi /etc/profile

b.sudo vi ~/.bashrc

在文件最后一行,添加

export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/usr/local/cuda/lib64/

添加完之后,需要进行更新:

a.source /etc/profile

b.source ~/.bashrc

进行重启之后,测试是否安装成功

cd ~/NVIDIA_CUDA_10.1_Samples/1_Utilities/deviceQuery

make -j4

sudo ./deviceQuery

若有输出相关信息,result=pass,表面安装完成。

5.安装cudnn

查看相应版本,并在官网进行搜索,最后点击如图红框。

ubuntu16.04配置显卡驱动和CUDA,CUDNN

我是通过windows远程连接的,下载下来为.solitairetheme8格式文件,将其传入服务器目录,进行转换格式和解压。

cp cudnn-10.1-linux-x64-v7.6.3.30.solitairetheme8 cudnn-10.1-linux-x64-v7.6.3.30.tgz

tar -xvf cudnn-10.1-linux-x64-v7.6.3.30.tgz

再将解压后的文件拷贝到相应路径下:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

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

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/lcoal/cuda/lib64/libcudnn*

没有提示错误,就表明安装完成。 

 

 

 

 

 

 

 

 

 

相关文章: