Pycharm配置远程pytorch服务器
Pycharm配置远程pytorch服务器
准备工作
- PyCharm专业版
- 配置好PyTorch的服务器
开始配置
-
依次点击按钮Tools->Deployment->Configuration
-
在弹出的界面中点击加号添加sftp,配置要求很简单,其中Web server URL可以不填
-
回到PyCharm初始界面
-
依次点击按钮Settings->Project->Python Interpreter
-
在弹出的界面中点击小齿轮中的Add,找到SSH Interpreter,依次填入,最终输入解释器在服务器中的位置
-
完毕
尝试
我们以PyTorch官方教程中的cifar10分类代码为例进行尝试
链接地址:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html?highlight=cifar
注意:如果你的服务器显卡过多的话,还需要在Run->Edit Configurations,找到Environment variables,输入CUDA_VISIBLE_DEVICES=0,1,2,3
,其中用分号隔开即可。
运行结果
总结
- PyCharm作为一款功能强大的Python IDE,由JetBrains打造,可以帮助我们提高写代码效率
- 就本篇而言,当模型较大的时候,程序一跑就要跑一天,本机电脑由于突发状况需要关机的话,程序就会中断。从这点出发,上面的配置就显得不是很友好了。推荐使用SSH工具加tmux软件进行深度学习代码编写,并且VSCode也有很多sftp的插件,也能有很高的效率,但是整体性确实还是PyCharm好一些。