深度学习环境之Pytorch和torch_Geometric的搭建
1. Pytorch 和 PyG
Pytorch和PyG是深度学习的python环境,后者用于图处理。其主要特点如下:
①需要与cuda(NVIDIA)配合,必须先装载cuda。
②cuda,pytorch,pytorch_geometric三者之间版本依赖精密,必须属于同一套版本。
其安装顺序即 cuda > pytorch > PyG
③其不能通过pycharm进行package安装,必须先安装在环境中,再由pycharm关联使用该环境。
④推荐使用anoconda创立环境并下载如上两包。
2. 配置步骤
2.1 cuda的安装
CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
详见如下博客:
https://blog.csdn.net/NSJim/article/details/115408138
2.2 anoconda 环境激活
使用Anoconda prompt(windows内搜索)进入anoconda下的命令行,据此创建/激活环境,并进行导包等操作。
详见链接:
https://blog.csdn.net/SARACH_WONG/article/details/89328307
2.3 pytorch的安装
注:pytorch对cuda的版本是向下兼容的,即可以使用更低等级的cuda,如11.1的cuda,可以安装对应10.2版本cuda的pytorch。
- 进入Pytorch官网(https://pytorch.org/),选择对应的版本号(可向下兼容),这里我们选择稳定版本1.10.2和cuda10.2
- 将图示中Run this Command所提供的代码在Anoconda激活的环境中运行(即
activate phm2012
phm2012为环境名),安装pytorch。
2.4 Pytorch_Geometri的安装
同理,基于上述基础,在Anoconda中进行PyG的导入:
- 进入PyG官网(https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html)
- 选择版本号,在Anoconda激活的环境中安装。
2.5 Pycharm中找到该环境
在pycharm中找到配置好的anoconda环境即可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人