系统环境: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即可
系统环境: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
网友评论