Advertisement

安装torch-geometric的步骤

阅读量:

文章结构概览

  • 零。安装步骤
    • 一、其他问题

      • 2.1
      • 2.2
      • 2.3

torch-geometric:一个用于图神经网络的工具包。

零。安装步骤

官方链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
1,创建新的conda环境(可选)

复制代码
    conda create -n geo1 python=3.6
    conda activate geo1
    conda deactivate
    
    
      
      
      
    
复制代码
    nvcc --version	
    
    
      
    

10.0
所使用的nvcc版本为10.0,对于那些正在寻找torch 100相关资源的用户而言,Geometric库似乎仅兼容torch 1.4与torch 1.5这两个版本。

2, 安装pytorch及相应的驱动程序
pytorch官网

复制代码
    pip insta

全部评论 (0)

还没有任何评论哟~