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

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
和网络发展的快,获取信息太容易了;另一方面是个人成长经验也丰富了,遇到事直到怎么解决,方向明确,报错就解决错,一个个解决了自然就得到想要的结果了,所谓事缓从恒,事急从权;事缓则圆,事急则乱。


http://www.kler.cn/a/563808.html

相关文章:

  • <02.26>Leetcode
  • 航旅纵横测试开发一面面经
  • PyTorch下三角矩阵生成函数torch.tril的深度解析
  • Python 高级特性-迭代器
  • 机器学习01
  • servlet相关
  • 【漫画机器学习系列】102.带泄露线性整流函数(Leaky ReLU)
  • playwright GitHub Actions运行测试
  • Fetch 是用于发起HTTP请求的API body 部分
  • 服务器主板可以单独升级吗?有什么影响?
  • 超过DeepSeek、o3,Claude发布全球首个混合推理模型,并将完成新一轮35亿美元融资...
  • 上海商米科技通信工程师后端开发岗内推
  • 从 0 到 1:使用 Docker 部署个人博客系统
  • 【Python爬虫(88)】当Python爬虫邂逅智能硬件:解锁数据新玩法
  • git设置本地代理
  • IO 和NIO有什么区别?
  • 科技项目查新指南:流程要点与材料准备
  • 比较RPC和RESTful API的优缺点
  • 力扣1210. 穿过迷宫的最少移动次数
  • 从0到一实现React Fiber从零到一实现React Fiber