windows 安装conda+python+pytorch+pyg+ogb指定版本(最全面)
conda创建虚拟环境
例如环境名字为neocpu
conda create -n neocpu python=3.8.8
如果遇到镜像源问题,404则删除镜像源
删除conda镜像源
conda config --remove-key channels
安装pytorch
进入创建的环境
activate neocpu
进入pytorch官网 找具体要下载的版本,我这里是1.8.1
查看电脑是不是带gpu,带的话查看cuda版本
nvcc -V
我这里直接使用的cpu
安装pyg2.0.3
试了很多方法,只有离线安装是可以的
由于我这里pytorch版本是1.8.1
例如是pytorch1.8.1和cuda10.2版本就访问
https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
访问
我这里是cpu和pytorch1.8.1就访问下面
https://pytorch-geometric.com/whl/torch-1.8.1+cpu.html
找到你要下载的离线轮子
一般先要下载4个依赖库
torch-scatter
torch-sparse
torch-cluster
torch-spline-conv
根据你下载的python版本和操作系统下载对应的轮子
我这里要下载pyg2.0.3,所以四个依赖文件全都下载最新的就行
库 | 版本 |
---|---|
torch-scatter | 2.0.8 |
torch-sparse | 0.6.12 |
torch-cluster | 1.5.9 |
torch-spline-conv | 1.2.1 |
下载完之后,进入下载文件的目录
离线安装
pip install XXX.whl
最后安装你想要的pyg版本
pip install torch-geometric==2.0.3
安装ogb
pip install ogb==x.x.x