概述
在编写常规深度学习代码时,总是需要使用服务器上的GPU资源,所以一般要写完代码,放到服务器,然后使用GPU运行。但是由于之前的习惯,总想本地调试一下或者本地直接跑测试结果,再放到服务器去跑。
网上查了一些文档,遇到了一系列问题,最终还是完美的解决了。整理一下分享出来希望对大家有用。
文档包括三个方面:
- 服务器Docker启动,在Docker中使用GPU资源
- Docker开启ssh登录
- PyCharm连接远程服务器Docker,使用python解释器和远程GPU
1. Docker中使用GPU
我这里使用的镜像是 registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.3.0-py38-torch1.11.0-tf1.15.5-1.6.1
这个影响不大。
为什么要使用Docker呢,因为这样能保证开发环境的一致性。
启动命令:
docker run -p 2017:22 --name="gpu-test" --gpus all -it -v /root/localpath:/workspace registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.3.0-py38-torch1.11.0-tf1.15.5-1.6.1 bash
-p 2017:22
: 映射了一个宿主机的2017端口到容器的22端口,为了之后连接容器
-v /root/localpath:/workspace
: 加了映射容器的磁盘到宿主机上
这样启动后,容器中就可以看到所有的GPU资源
可以验证下是否可以使用gpu
import torch
print(f'\nAvailable cuda = {torch.cuda.is_available()}')
print(f'\nGPUs availables = {torch.cuda.device_count()}')
print(f'\nCurrent device = {torch.cuda.current_device()}')
print(f'\nCurrent Device location = {torch.cuda.device(0)}')
print(f'\nName of the device = {torch.cuda.get_device_name(0)}')
这里需要注意宿主机是有GPU驱动的,容器中有cuda等环境。
到这个阶段,就可以在容器中跑程序了,如果没有本地调试需求,可以把代码传到宿主机目录,在容器中可以直接运行。如图:
2. 开启Docker中ssh
因为后续需要PyCharm连接Docker容器,所以容器中需要现有ssh服务,并把22端口暴露出去,我们是映射到宿主机上一个端口上。
docker run -p 2017:22
在我提供的这个镜像上没有ssh,需要手动安装一下,如果有ssh就不需要装了。
安装脚本:
# 更新密码
passwd# 更新源
apt-get -y update# 安装sshserver
apt-get install openssh-server
apt-get install openssh-client# 开启密码登录
vim /etc/ssh/sshd_config
在文件最后加上一行
PermitRootLogin yes #允许root用户使用ssh登录# 重启ssh服务
/etc/init.d/ssh restart# 尝试连接远程容器
ssh root@10.22.33.44 -p 2017
3. PyCharm中连接Docker中的Python解释器和GPU
有两个地方需要配置,但是可以一步配置完成。
我使用的PyCharm 2023.2.1 (Professional Editor)版本
使用远程解释器的操作步骤:
-
创建Interpreter
操作路径:Add New Interpreter --> On SSH -
填写远程服务器上Docker容器连接信息
这里弹框需要填入远程信息,这里的IP是远程宿主机,端口是容器暴露的2017端口,2017会映射到容器内部22端口上,就可以通过2017连接到容器22. -
输入密码
-
设置解释器地址
这一步需要特别注意
。
第一Environment
: 一定选择远程已存在的解释器,不要创建新的解释器。
第二Interpreter
: 解释器地址,选择远程容器中的解释器地址,具体位置可以在远程中查询,查询方式:
进入python
import os
print(os.sys.executable) # 会输出解释器地址# 我使用的镜像的解释器位置,所以我在这一步设置的这个地址
/opt/conda/bin/python
第三sync folders
: 是选择把本地文件同步到远程容器的什么目录。这儿可以选择一个容器的挂载目录,防止容器重启,远程文件丢失,还需要重传。
最后自动上传可以勾选上,修改后会自动上传。
创建完成后,可以在interpreter中查看,依赖是否跟远程相同。
如果显示的远程pip list中依赖,即成功。
然后执行Run试试。
参考链接:https://zhuanlan.zhihu.com/p/605389180