如此不智能不便利不友好的库竟然还有人喜欢用。
不会吧不会吧不会吧不会吧

记录傻X环境依赖。

torch1.2.0 torchvision0.4.0 cuda10.0 3.5<=python<=3.7
torch1.3.0 torchvision0.4.1 cuda10.0 3.5<=python<=3.7
torch1.3.1 torchvision0.4.2 cuda10.0 3.5<=python<=3.7
torch1.4.0 torchvision0.5.0 默认cuda10.1(支持10.0) 3.5<=python<=3.8
torch1.5.0 torchvision0.6.0 默认cuda10.1 python>=3.6
torch1.6.0 torchvision0.7.0 默认cuda10.1 python>=3.6
torch1.7.0 torchvision 0.8.0 默认cuda10.2 python>=3.6
torch1.7.1 torchvision0.8.2 默认cuda10.2 python>=3.6

如果装的是cu10.0，只能用torch1.4之前的低版本了。
torch版本太高了会报错

OSError: libcusparse.so.10: cannot open shared object file: No such file or directory

因为torch里面用的是10.1，但我本地是10.0，找不到这个文件。

查询torch里面用的版本

python -c 'import torch; print(torch.__version__)'
python -c 'import torch; print(torch.version.cuda)'

打印结果是

10.1

你看看这个东西，
一个库弄出4个库来。

pip install torch-sparse
pip install torch-scatter
pip install torch-cluster
pip install torch-geometric
pip install torch-spline-conv(optional)

官网

https://pytorch-geometric.com/whl/

装geometric依赖于torch版本。
torch版本决定了torchvision版本。
torchvision版本疑似决定了默认使用的cudatoolkit版本。
所以最终geometric需要跟cuda版本对应。

如果是cuda10.0！只能用torch1.4.0！还要指定cuda版本，不然默认给你装10.1。（21.04.15）

pip install torch1.4.0+cu100 torchvision0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

文件链接

https://download.pytorch.org/whl/cu100/torch-1.4.0%2Bcu100-cp37-cp37m-linux_x86_64.whl
https://download.pytorch.org/whl/cu100/torchvision-0.5.0%2Bcu100-cp37-cp37m-linux_x86_64.whl

配合torch1.4.0+cuda10.0+python3.7装geometric。

https://pytorch-geometric.com/whl/torch-1.4.0+cu100.html

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.4.0+cu100.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.4.0+cu100.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.4.0+cu100.html
pip install torch-geometric

上面不行就本地安装

wget https://pytorch-geometric.com/whl/torch-1.4.0+cu100/torch_cluster-1.5.4-cp37-cp37m-linux_x86_64.whl
wget https://pytorch-geometric.com/whl/torch-1.4.0+cu100/torch_scatter-2.0.4-cp37-cp37m-linux_x86_64.whl
wget https://pytorch-geometric.com/whl/torch-1.4.0+cu100/torch_sparse-0.6.1-cp37-cp37m-linux_x86_64.whl
pip install *.whl
pip install torch-geometric

配合torch1.7.0+cuda11.0

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-geometric

卸载

pip uninstall torch-scatter
pip uninstall torch-sparse
pip uninstall torch-cluster
pip uninstall torch-spline-conv
pip uninstall torch-geometric

其他版本

跑geometric的开源项目，经常离不开ogb数据包。
截止21.04最新的ogb包是1.3.1版本的。
然而ogb1.3.1要求torch>=1.6.0。
所以我们为了适配cuda10.0，torch1.4.0。
只能降格使用ogb1.2.0，（torch>=1.2.0)

如果还要用dgl库(目前最前0.6.1)
推荐降格使用dgl==0.4.3

补充
遇到nvcc执行失败

vim ~/.bashrc

修改

export CUDA_HOME=/usr/local/cudaexport PATH="/usr/local/cuda-10.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH"

中间文件夹的名字跟cuda版本对应。
参考的是官方提供的解决方案

https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html#installation-via-binaries

更新 torch1.9.0 + cuda11.1 + pyg2.0.3

torch 1.10 , cuda 10.2 python 3.7.9
torch-geometric 2.0.2

pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0%2Bcu102/torch_sparse-0.6.12-cp37-cp37m-linux_x86_64.whl
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0%2Bcu102/torch_scatter-2.0.9-cp37-cp37m-linux_x86_64.whl
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.10.0%2Bcu102/torch_cluster-1.5.9-cp37-cp37m-linux_x86_64.whl

配合torch 1.9.0 + cuda11.1 + python3.9.7

pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch-geometric==2.0.3
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0%2Bcu111.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0%2Bcu111.html
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0%2Bcu111.html

更新 torch 1.11.0 + cuda11.3 + pyg2.0.4

torch1.11.0

https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp39-cp39-linux_x86_64.whl

有两个源路径

https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu113.html
https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu115.html

pip install torch-geometric==2.0.4
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu113.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu113.html
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu113.html

对cuda10.2 ，用torch1.11 + pyg2.0.4+ torch-sparse0.6.13有问题。
import torch_geometric.nn 会报 segmentfault
回退
torch 1.10.2
torch-geometric 2.0.3
torch-scatter 2.0.9
torch-sparse 0.6.12
torch-cluster 1.5.9

折腾torch geometric环境相关推荐

安装caffe-yolov3时报错：ImportError: No module named torch（环境变量设置）
linux上安装完pytorch后仍报错: Traceback (most recent call last):File "cifar10_tutorial.py", line 5 ...
rf运行python脚本报错_GNUradio与USRP折腾指南之环境搭建
其实这篇文章我想用摩斯电码写,但是由于我不会,所以我只能敲汉字了. 我五月份的时候拿到了两台USRP,是由国内山黑制作的.没想到居然出差一个多月,耽误了很多事情,现在我已经返回到我的家中,在折腾了很长 ...
这可能是我觉得最折腾的C++环境配置（编码+调试）--mac+eclipse
本着造福大众的心态,万一可能有人喜欢这种环境开发呢对吧~~折腾了一天,又是谷歌又是百度,讲真遇到报错问题搜出的资料挺少的,看来用这类环境开发C++的人不多,毕竟谁没事干放着xcode,vscode不用 ...
下载drivegoogle文件；cmd安装出现黄色警告；DGL下载；torch版本；(最全)GPU-torch下载；WARNING: Retrying Retry(total=4, connect报错
Linux服务器用gdown下载google Drive中文件_栗子菜菜的博客-CSDN博客终端环境如何下载谷歌网盘google drive文件_Neil-Yale的博客-CSDN博客下载goog ...
超详细配置教程，搭建 Windows 深度学习环境
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达选自 | towardsdatascience 作者 | Ahina ...
Miniconda3+Tensorflow2.3(GPU版)+Win10_x64+GTX1060深度学习环境搭建
写在这里的初衷,一是备忘,二是希望得到高人指点,三是希望能遇到志同道合的朋友. 硬件信息: 系统:win10家庭中文版 CPU:i7-7700HQ 内存:16GB 显卡:GTX1060 目录一.确定 ...
Miniconda3+PyTorch1.7.1(GPU版)+Win10_x64+GTX1060深度学习环境搭建
写在这里的初衷,一是备忘,二是希望得到高人指点,三是希望能遇到志同道合的朋友. 硬件信息: 系统:win10家庭中文版 CPU:i7-7700HQ 内存:16GB 显卡:GTX1060 目录一.确定 ...
windows上安装gcc/g++环境（MinGW，msys64等）
1 前言经过折腾安装gcc环境,记录一下 2 工具 MinGW安装器: 下载地址1:MinGW官网下载地址2:SourceForge (MinGW w64) 下载地址3:win-builds-1 ...
Ubuntu 下一个 vim 建立python 周围环境构造
于Windows通过使用各种现成的工具使用,去Linux下一个,没有一个关于线索--总之google有些人的经验,折腾来折腾,开发环境也算是一个好工作. 1. 安装完成vim # apt-get in ...

折腾torch geometric环境

更新 torch1.9.0 + cuda11.1 + pyg2.0.3

更新 torch 1.11.0 + cuda11.3 + pyg2.0.4

折腾torch geometric环境相关推荐

最新文章

热门文章