我要评论linux部署pytorch
背景介绍
不同的开源代码可能需要不同的实验环境和版本，这时候的确体现出anaconda管理环境的好处了，分别搞一个独立环境方便管理。
有的教程建议选择较旧的版本，但笔者建议在条件允许的情况下安装最新版，本次是打算使用扩散生成库diffuser库，但在torch1.13+cu117版本下会报类型错误，多亏有ai，很快定位到了是torch版本过低问题，但更新的torch也需要更高版本的cuda，因为服务器不是笔者自己在用，贸然升级cuda可能爆发组内矛盾，所以顺路也解决了虚拟环境内如何安装cuda的问题。
anaconda安装
本次服务器上已有anaconda，有关方法可见linux服务器上安装anaconda，大致步骤是到官网查找需要的版本连接，使用wget方法下载，安装后添加环境变量。
linux安装pytorch
剩下方法与windows中安装pytorch几乎完全一样，conda create -n 环境名 python=3.9创建conda环境后，到pytorch官网找合适的版本下载。
安装时注意版本即可，服务器虚拟环境中输入命令后即可顺利安装。
虚拟环境安装cuda
参照安装cudatoolkit，首先使用conda actviate 环境名进入先前创建的环境，出现下图前缀变换说明切换成功：
先使用nvidia-v查看本机的显卡信息：
后续下载安装的cuda版本不能高于该cuda版本，使用conda search cudatoolkit查找可以安装的cuda版本：
查找到想要的版本后，直接在当前环境使用conda install cudatoolkit=11.8.0安装即可。
到此服务器的虚拟环境中已具有pytorch+cuda全部所需条件。
pycharm连接服务器
linux系统无可视化界面，更无编程所需的ide，使用ide远程连接服务器编程十分必要，实现的方式也多种多样。
文件选项内远程开发就有选项，通过ssh连接
但该方法笔者还没成功，涉及证书问题，不过发现了另外一种方法，可用ssh连接远程服务器的解释器，使用如下步骤即可：
大致原理为使用ssh协议将本地文件上传到服务器上，服务器执行文件后结果返回给pycharm，服务器端的文件位置可通过同步文件夹选项设置。
执行代码
import torch print(torch.__version__)
得到如下输出说明实验成功：
总结
以上就是pycharm远程连接服务器运行pytorch的过程详解的详细内容，更多关于pycharm远程连接运行pytorch的资料请关注代码网其它相关文章！
发表评论