Ubuntu系统中使用pycharm远程训练pytorch模型。(服务器租用autodl)

1.安装pycharm专业版

可以用学生邮箱申请pycharm专业版。具体可以参考https://zhuanlan.zhihu.com/p/163521228

2.用filezilla上传数据集。

不要用pycharm上传所有数据,pycharm上传慢而且容易失败。亲测3GB的数据集传了6个小时还没传好,还失败了一些文件。最好使用filezilla上传数据集。

3.连接远程服务器

这个可以看autodl的帮助文档https://www.gpuhub.com/docs/pycharm/

这里要注意一点:host后面把空格删掉!这里连接的时候注意选择Python环境的时候一定要选择existing environment。**选择system interpreter,不要选择虚拟环境!。其他的按照帮助文档的选择就可以了。**连接完成之后,pycharm会把project传上去。不要把数据 集放在project里,pycharm会把project里所有的文件传上去,而且你无法取消。

Ubuntu系统中使用pycharm远程训练pytorch模型。(服务器租用autodl)_第1张图片

4 run

点击run运行模型
如果报错,看看错误类型,一般如果是缺少包的话,直接在远程终端中pip install “包的名称”来安装。

5监测GPU的用量

在终端中输入nvidia-smi

Ubuntu系统中使用pycharm远程训练pytorch模型。(服务器租用autodl)_第2张图片

第一个是显存用量,第二个是显卡使用率。 

感谢Bubbliiiing大神提供的YOLO模型和数据集。

项目gethub地址是http://Access has been restricted

你可能感兴趣的:(ubuntu,pycharm,pytorch,神经网络,深度学习)