linux服务器搭建

背景

在服务器上搭建docker容器，并远程ssh连接。在本服务器上不运行代码，可以不安装cuda和cudnn等，保证安装nvidia显卡驱动，就可以在容器中安装cuda调用nvidia显卡驱动。由于可以拉取带有cuda的镜像，因此不需要手动安装cuda等。本文主要讲述在安装好系统和显卡驱动的前提下，进行服务器的建立容器并远程ssh的操作。

前提

已经成功安装linux系统和nvidia显卡驱动

sudo nvidia-smi

安装docker

1、下载安装包

打开终端，输入如下命令：

wget -qO- https://get.docker.com/ | sh

2、更新docker用户组

打开终端，输入如下命令：

sudo groupadd docker          
sudo gpasswd -a $XXX docker   
sudo gpasswd -a $USER docker  
newgrp docker                 

3、测试安装成功

输入如下命令：

docker run hello-world

输出如下内容，则安装成功：

Hello from Docker!
This message shows that your installation appears to be working correctly.
 
To generate this message, Docker took the following steps:
 1. The Docker client contacted the Docker daemon.
 2. The Docker daemon pulled the "hello-world" image from the Docker Hub.
    (amd64)
 3. The Docker daemon created a new container from that image which runs the
    executable that produces the output you are currently reading.
 4. The Docker daemon streamed that output to the Docker client, which sent it
    to your terminal.
 
To try something more ambitious, you can run an Ubuntu container with:
 $ docker run -it ubuntu bash
 
Share images, automate workflows, and more with a free Docker ID:
 https://hub.docker.com/
 
For more examples and ideas, visit:
 https://docs.docker.com/get-started/

安装nvidia-docker

1、下载安装包

输入以下命令：

curl https://get.docker.com | sh
sudo systemctl start docker
sudo systemctl enable docker

2、设定稳定存储库和GPG密钥

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

3、安装

更新包列表后安装，输入以下命令：

sudo apt-get update
sudo apt-get install -y nvidia-docker2

重启docker，输入以下命令：

sudo systemctl restart docker

4、创建容器并pull镜像

新建一个名为develop的docker，同时安装nvidia/cuda:11.4.2-devel-ubuntu20.04，将develop容器的‘home‘映射到本地文件夹‘/data/code:/home‘，设计自启动，端口映射（例如将develop的22端口映射到本地的224，需要注意的是本机的224端口用于ssh连接）nvidia docker 仓库：https://hub.docker.com/r/nvidia/cuda/tags，可以根据情况选择，此处为‘nvidia/cuda:11.4.2-devel-ubuntu20.04。值得注意的是，base代表基础版，大多情况下选择安装devel。输入以下代码：

sudo docker run -it --ipc=host --gpus all -v /data/code:/home --restart=always --name develop  -p 225:22 nvidia/cuda:11.4.2-devel-ubuntu20.04

5、进入docker

输入以下代码：

docker attach develop # develop为本文新建的docker名称

6、后台运行docker，输入以下代码：

ctrl+p+q

在容器内配置ssh

1、安装ssh

输入如下代码：

sudo apt-get update
sudo apt-get install -y openssh-server
sudo apt-get update
sudo apt-get install ssh

2、启动ssh

/etc/init.d/ssh start

输出以下内容代表启动成功

3、查看ssh状态

输入以下代码：

ps -e|grep ssh

有如下输出代表成功

4、修改ssh密码

输入以下代码：

passwd root

按照提示修改密码：

5、修改参数文件，使root允许被访问

apt-get install vim
vim /etc/ssh/ssh_config

改为passwordAuthentication yes:

vim /etc/ssh/sshd_config

改为PermitRootLogin yes:

5、重启ssh

/etc/init.d/ssh restart

6、查看ip地址

退出容器，并查看ip地址

exit
apt-get install net-tools
ifconfig

最后，根据ip地址和端口号进行ssh连接即可！

ssh 到指定端口

ssh -p xx user@ip 

xx 为 端口号 user为用户名 ip为要登陆的ip