Py学习  »  docker

PyCharm+Docker:打造最舒适的深度学习炼丹炉

江大白 • 4 月前 • 91 次点击  

以下章来源于知乎:小白学视觉

作者:刘震

链接:https://zhuanlan.zhihu.com/p/52827335

本文仅用于学术分享,如有侵权,请联系台作删文处理

导读
本教程旨在帮助炼丹的同学们,通过PyCharm搭建本地编辑器与远程服务器的愉快工作环境。相较于简单的SFTP插件,PyCharm提供更强大的功能和更流畅的体验,帮助你更顺畅地在本地编辑器中使用远程服务器的环境,提高工作效率。

一般炼丹都在服务器上,很少有人在本机跑代码的。之前自己捣鼓怎么用本地的编辑器配上远程服务器的环境来营造一个最舒乎的姿势。最开始用VS Code的SFTP插件,但是用着还是不顺手,因为不能用远程的环境,最后还是选择了强大的PyCharm。下面的这个教程最开始只是自己用,后来写了一篇博客分享给实验室的人,没想到大家都开始用上了。这次专门写一篇更详细的放到知乎供有缘人使用。

开始之前你可以对比以下条件来确定自己是否真的需要这样做:

  • 使用PyCharm作为主力IDE写Python而不是其他(我非常推荐大家用)
  • 深度学习炼丹师(当然你可以不是,但如果是,你会更舒服)
  • 服务器在远程或者局域网(局域网的体验最佳)
  • 使用Docker(能够熟练使用最佳)

当你配置完成之后,你将能在本地PyCharm IDE上使用远程服务器的Python解释器和环境进行调试和运行,同时能够很方便的将代码在本地和服务器上进行同步。

OK,废话不多说,正式开始我们的PyCharm+Docker的远程环境配置之旅。

首先你需要满足以下两个必备条件:

  • 使用PyCharm专业版,记住一定是专业版(社区版不提供远程服务)
  • 在服务器上安装docker和nvidia-docker

01

新建docker container

首先按照如下命令新建一个docker容器(关于docker的使用这里不废话,建议不熟悉的童鞋先去学一下如何使用docker,教程很多):

sudo nvidia-docker run -it -p [host_port]:[container_port](do not use 8888) --name:[container_name] [image_name] -v [container_path]:[host_path] /bin/bash

举个栗子:

sudo nvidia-docker run -p 5592:5592 -p 5593:5593 -p 8022:22 --name="liuzhen_tf" -v ~/workspace/liuzhen/remote_workspace:/workspace/liuzhen/remote_workspace -it tensorflow/tensorflow:latest-gpu /bin/bash

正确执行完之后,现在我们就处在新建的docker容器里了(端口映射,容器名,镜像和路径映射这些换成你自己的就行,但是一定要留一个端口映射到容器22端口,因为SFTP默认使用22端口)。


02

配置SSH服务

接着我们在刚刚新建的容器里配置SSH服务,首先安装openssh-server:

$ apt update$ apt install -y openssh-server
然后建立一个配置文件夹并进行必要的配置:
$ mkdir /var/run/sshd$ echo 'root:passwd' | chpasswd# 这里使用你自己想设置的用户名和密码,但是一定要记住!$ sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config$ sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd$ echo "export VISIBLE=now" >> /etc/profile
如果连不上SSH服务,可能是某些版本的PermitRootLogin yes默认被注释了,可以使用如下(感谢 @Chenjie Xing 的反馈):
sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
重启SSH激活配置:
$ service ssh restart
在服务器(宿主机)上(不是服务器的docker里)测试刚刚新建docker容器中哪个端口转发到了服务器的22端口:
$ sudo docker port [your_container_name] 22# 如果前面的配置生效了,你会看到如下输出# 0.0.0.0:8022
最后测试能否用SSH连接到远程docker:
$ ssh root@[your_host_ip] -p 8022# 密码是你前面自己设置的
到这里说明服务器的docker端已经完成配置。


03

在PyCharm里配置部署环境

打开PyCharmTools > Deployment > Configuration, 新建一个SFTP服务器,名字自己取:

输入如下图配置,注意这里的端口是你刚刚设置的映射到容器22端口的宿主机中的端口,我这里使用的是8022,账号密码是你刚刚自己设置的,这里的Root Path设置一个远程docker容器里的路径:
配置完点击Test SFTP connection,如果成功就恭喜你,可以进行下一步了。
最后在Mappings中配置路径,这里的路径是你本地存放代码的路径,与刚刚配置的Root Path相互映射(意思是Mapping里本机的路径映射到远程的Root Path),方便以后在本地和远程docker中进行代码和其他文件同步。


04

在PyCharm里配置远程解释器
点击PyCharm的File > Setting > Project > Project Interpreter右边的设置按钮新建一个项目的远程解释器:
配置完成以后在项目解释器界面就会出现如下图所示,可以看到此时已经完成远程解释器的本地化:
配置完成以后需要等本地和远程的环境同步一下,到这里,恭喜你,可以用最舒服的姿势。。。写代码了。
配置完成以后的日常是这样的:
左边是本地的文件,修改之后可以随时右键deployment->upload到远程主机,或者直接在本地调试运行;最右边是远程主机的文件,假如直接在远程修改了文件刷新一下同样可以右键下载到本地,但是我不建议这样做,因为这样很容易带来冲突(毕竟没有很好的版本控制)。目前最好的实践是在局域网的服务器上,时延低,同步速度快。

05

常见问题
1. 最常见的问题就是docker容器停了以后里面的SSH服务也会相应停止,因此当你发现自己某一天连不上的时候,记得去docker里重启一下ssh服务:



    
$ service ssh restart
感谢 @丁果 对这个问题提出的解决方案:

“可以在run的时候加上--restart=always,同时ssh启动的写到dockerfile里去。”

2. 在容器中执行apt update的时候可能会出现0% working 的问题

感谢 @奥古斯都 提出该问题以及解决方案:

”这里不是源的问题,因为容器环境太过纯净,这里需要安装apt-transport-https这个deb文件,下载的时候也要注意不要下载最新的版本,否则也会出现依赖问题,要下载和当前docker容器内的apt相匹配的版本。“

针对粉丝,提供各研究领域的百节公开课免费领取,扫码回复客服:【你的研究方向】
根据不同级别论文,提供一对一指导:

(1)EI会议(核心)、EI期刊、SCI四区;

(2)人工智能领域毕业论文,SCI三区CCF,C类;

(3)SCI二区/CCFB类;SCI一区/CCFA类/中科院;
另支持指定期刊会议;
可指导方向:

机器学习、深度学习、计算机视觉、自然语言处理、多模态、推荐系统、时间序列、机器人 数据挖掘、强化学习、图神经网络,以及AI+安全/医学/金融/生物/交通/等各方向论文;

推荐阅读

AIHIA | AI人才创新发展联盟2023年盟友招募

AI融资 | 阿加犀智能科技获得A+轮近亿元融资

AI项目应用  | 万字长文教你一步步实现离岗识别

注意:人工智能行业,从算法研究、算法适配到项目落地,有很多流程。
为了便于大家学习交流,大白创建了一些不同方向的行业交流群
目前主要开设:AI视觉软件应用群、AI训练营开发者群企业级模型适配群,AI交流群
大家可以根据自己的兴趣爱好,加入对应的微信群,一起交流学习!
此外大白和数十名志同道合的AI小伙伴,一起筹备创建了AIHIA人才联盟。盟网站:www.aihiamgc.com
目前已有200+名盟友,已举办30多场技术主题分享及行业圆桌论坛,以及300多场线上会议,也欢迎大家加入。
© THE END 


大家一起加油! 

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/166304
 
91 次点击