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即可

上一篇下一篇

猜你喜欢

热点阅读