Pycharm配置远程pytorch服务器

Pycharm配置远程pytorch服务器

准备工作

1. PyCharm专业版
2. 配置好PyTorch的服务器

开始配置

1. 依次点击按钮Tools->Deployment->Configuration

2. 在弹出的界面中点击加号添加sftp，配置要求很简单，其中Web server URL可以不填

3. 回到PyCharm初始界面

4. 依次点击按钮Settings->Project->Python Interpreter

5. 在弹出的界面中点击小齿轮中的Add，找到SSH Interpreter，依次填入，最终输入解释器在服务器中的位置

6. 完毕

尝试

我们以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，其中用分号隔开即可。

运行结果

总结

1. PyCharm作为一款功能强大的Python IDE，由JetBrains打造，可以帮助我们提高写代码效率
2. 就本篇而言，当模型较大的时候，程序一跑就要跑一天，本机电脑由于突发状况需要关机的话，程序就会中断。从这点出发，上面的配置就显得不是很友好了。推荐使用SSH工具加tmux软件进行深度学习代码编写，并且VSCode也有很多sftp的插件，也能有很高的效率，但是整体性确实还是PyCharm好一些。