Ubuntu配置Pytorch on Graph (PoG)环境过程图解
最近准备复现一下 KDD-20 Towards Deeper Graph Neural Networks 的代码,顺便学习一下 GCN 最新的一些进展。 GCN 的代码通常需要安装 Pytorch on Graph, 按照其 官方指导, 发现无法兼容当前所使用的服务器上的 pytorch1.3 和 cuda10.0 环境, 需要升级 cuda 和 pytorch 的版本。 于是准备重新安装一个虚拟环境,使用上 cuda 10.2 和 pytorch1.6。 1. 安装 Cuda10.2 以及对应版本的 Cudnn 安装 cuda 10.2, wget http://developer.dow
用户评论