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
查看相应版本,并在官网进行搜索,最后点击如图红框。
我是通过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*
没有提示错误,就表明安装完成。