当前位置: 首页 > news >正文

pycharm远程连接服务器运行pytorch

Linux部署pytorch

背景介绍

不同的开源代码可能需要不同的实验环境和版本,这时候的确体现出Anaconda管理环境的好处了,分别搞一个独立环境方便管理。

有的教程建议选择较旧的版本,但笔者建议在条件允许的情况下安装最新版,本次是打算使用扩散生成库diffuser库,但在torch1.13+cu117版本下会报类型错误,多亏有ai,很快定位到了是torch版本过低问题,但更新的torch也需要更高版本的cuda,因为服务器不是笔者自己在用,贸然升级cuda可能爆发组内矛盾,所以顺路也解决了虚拟环境内如何安装cuda的问题。

Anaconda安装

本次服务器上已有Anaconda,有关方法可见Linux服务器上安装Anaconda,大致步骤是到官网查找需要的版本连接,使用wget方法下载,安装后添加环境变量。
下载Anaconda

Linux安装pytorch

剩下方法与Windows中安装pytorch几乎完全一样,conda create -n 环境名 python=3.9创建conda环境后,到pytorch官网找合适的版本下载。

pytorch安装
安装时注意版本即可,服务器虚拟环境中输入命令后即可顺利安装。

虚拟环境安装cuda

参照安装cudatoolkit,首先使用conda actviate 环境名进入先前创建的环境,出现下图前缀变换说明切换成功:
切换环境
先使用nvidia-v查看本机的显卡信息:
查看显卡信息
后续下载安装的cuda版本不能高于该CUDA版本,使用conda search cudatoolkit查找可以安装的cuda版本:
cuda版本
查找到想要的版本后,直接在当前环境使用conda install cudatoolkit=11.8.0安装即可。

到此服务器的虚拟环境中已具有pytorch+cuda全部所需条件。

pycharm连接服务器

Linux系统无可视化界面,更无编程所需的IDE,使用IDE远程连接服务器编程十分必要,实现的方式也多种多样。

文件选项内远程开发就有选项,通过ssh连接
远程开发
但该方法笔者还没成功,涉及证书问题,不过发现了另外一种方法,可用ssh连接远程服务器的解释器,使用如下步骤即可:
添加解释器
使用现有环境
大致原理为使用ssh协议将本地文件上传到服务器上,服务器执行文件后结果返回给pycharm,服务器端的文件位置可通过同步文件夹选项设置。

执行代码

import torch
print(torch.__version__)

得到如下输出说明实验成功:
实验输出

总结

这在本科阶段应该是得折腾个把礼拜的事,现在没费多少劲就解决了,一方面是现在AI
和网络发展的快,获取信息太容易了;另一方面是个人成长经验也丰富了,遇到事直到怎么解决,方向明确,报错就解决错,一个个解决了自然就得到想要的结果了,所谓事缓从恒,事急从权;事缓则圆,事急则乱。

相关文章:

  • web理论总结
  • MQTT实现智能家居------5、交叉编译
  • 《从零到全栈:CSS3新增属性》
  • 【每日八股】MySQL篇(三):索引(上)
  • 自动驾驶泊车算法详解(一)
  • Python使用列表实现栈、队列学习记录
  • macpro m1 安装deepseek
  • QEMU 的详细介绍、安装指南、配置说明
  • vuetify项目添加代理跨域请求
  • 【保姆级视频教程(二)】YOLOv12训练数据集构建:标签格式转换-划分-YAML 配置 避坑指南 | 小白也能轻松玩转目标检测!
  • 基于 ‌MySQL 数据库‌对三级视图(用户视图、DBA视图、内部视图)的详细解释
  • 2.2 添加注释
  • Uniapp 中布局魔法:display 属性
  • Redis速成(1)VMware虚拟机安装Redis+Session验证登录注册+MybatisPlus
  • 排序03(数据结构初阶)
  • 2.16作业
  • 数据库的常见权限作用
  • 在MAC上面通过HomeBrew安装node和npm@指定版本
  • Chart.js 折线图深入解析与使用指南
  • CentOS安装MariaDB
  • 微信公众平台人工电话/网站播放视频速度优化
  • 柳州网站建设价格/推广方式都有哪些
  • 网站怎么做才能被百度收录/百度推广电话客服
  • 胶州网站建设哪家好/沈阳网页建站模板
  • 网站制作公司很好 乐云践新/内容营销
  • 网站建设公司是干嘛的/一键建站