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

安装torch\torch-geometric

(1) cpu:

直接用以下代码顺利完成安装:

pip install torch== -i https://pypi.tuna.tsinghua.edu.cn/simple/

pip install torch-cluster torch-scatter torch-sparse -i https://pypi.tuna.tsinghua.edu.cn/simple/

(2)gpu:

按照以上的步骤没安装成功:

1、cuda版本和torch对应,但是pytorch2.0不支持geometric

torch-geometric 需要与 PyTorch 版本完全兼容,否则会出现不可预期的错误。由于 PyTorch 2.0 不再被支持,所以 torch-geometric 没有针对该版本进行开发和测试。 

pip install torch==<your_pytorch_version>
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-geometric==<your_torch_geometric_version>

其中 <your_pytorch_version><your_torch_geometric_version> 分别替换为你想要安装的 PyTorch 和 torch-geometric 版本号。

cu111 表示 CUDA 11.1 版本,如果你使用的是其他版本的 CUDA,需要对应修改。

2、建议换为torch2.0以下的进行安装:

先查看cuda 版本【这里我是11.7】

找对应的torch版本:

CUDA 11.7 支持的 PyTorch 版本包括:

  • PyTorch 1.10.0
  • PyTorch 1.9.0

其中,PyTorch 1.9.0 是一个支持 CUDA 11.1/11.2/11.3/11.4/11.5/11.6/11.7 的稳定版本。而 PyTorch 1.10.0 更是增加了对 CUDA 11.4/11.5/11.6/11.7 的原生支持。

注意:安装前先卸载之前安装的torch相关软件:

a、CUDA Toolkit 是 NVIDIA 提供的一套用于开发和运行 CUDA 应用程序的软件包。CUDA 是一种并行计算平台和编程模型,可以利用 GPU 加速各种应用程序的计算过程。

CUDA Toolkit 包含以下组件:

  1. CUDA 编译器:用于将 CUDA C/C++ 代码编译成可以在 GPU 上执行的二进制代码。

  2. CUDA 运行时库:包含 GPU 加速的数学函数库、通信库、并行算法等。

  3. CUDA 驱动程序:用于与 GPU 硬件进行交互,管理 GPU 内存等底层操作。

  4. CUDA 工具集:包括了 CUDA Profiler、CUDA Visual Profiler、CUDA-GDB 和 nvprof 等工具,用于分析和调试 CUDA 应用程序。

安装 CUDA Toolkit 可以使你的计算机支持 CUDA 技术,并且可以使用 CUDA 软件开发包(SDK)进行 GPU 加速的开发和优化。如果你想要在计算中使用 GPU 计算,建议先安装相应版本的 CUDA Toolkit,并确保你的计算机中有支持 CUDA 的 NVIDIA 显卡。



 

进行安装:

pip install torch==1.13.1+cu117 torchvision==0.10.0+cu116 torchaudio==0.9.0+cu116 -f https://download.pytorch.org/whl/cu117/torch_stable.html
ps:CUDA 11.7对应的cudatoolkit版本应该为11.3

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-<your_pytorch_version>+${CUDA}.html
pip install torch-geometric==<your_torch_geometric_version>


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

相关文章:

  • Raft协议
  • MFC - 控件的消息和控件的事件(命令)有什么区别?
  • leetcode单词的个数
  • 憨批的语义分割重制版11——Keras 搭建自己的HRNetV2语义分割平台
  • stm32 esp01s Qt 巴法云平台控制小灯
  • C++模板基础(六)
  • 故障定级和定责
  • 处理机调度与死锁习题
  • 蓝桥杯第十四届省赛完整题解 C/C++ B组
  • Window常用命令
  • Dubbo架构整体设计
  • 小规模容器编排使用Docker Swarm不香么,用个锤子的kubernetes
  • Leetcode.2399 检查相同字母间的距离
  • webpack介绍
  • 三十四、java中的引用
  • react简单实现防抖和节流教程方法
  • 用于平抑可再生能源功率波动的储能电站建模及评价(Matlab代码实现)
  • 【前端做项目常用】相关插件的官网 总结
  • 【web自动化测试】
  • LinuxGUI自动化测试框架搭建(七)-Ubuntu上安装配置Pycharm