torch-scatter、torch-sparse的版本问题
2019-11-29 本文已影响0人
SSRSGJYD
系统环境:ubuntu,CUDA 10.0, python 3.6
问题:在安装torch-scatter、torch-sparse的时候,发现在pytorch==1.3下,CUDA 10.0的版本不匹配(可能是需要升级到CUDA 10.1)。在pytorch==1.0的时候,torch-scatter、torch-sparse也无法成功编译。
解决方案:选择安装pytorch==1.1即可