美文网首页
torch-scatter、torch-sparse的版本问题

torch-scatter、torch-sparse的版本问题

作者: SSRSGJYD | 来源:发表于2019-11-29 17:43 被阅读0次

    系统环境: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即可

    相关文章

      网友评论

          本文标题:torch-scatter、torch-sparse的版本问题

          本文链接:https://www.haomeiwen.com/subject/jybgwctx.html