Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter

在win 10 环境下配置torch-scatter,经历了很多坑,最后总结起来,就是要看官方文档,先给出官方文档的连接及相关有用的链接:

https://pytorch.org/

https://github.com/rusty1s/pytorch_geometric

https://github.com/pytorch/pytorch

在win 10环境下,cuda的版本只能如下所示:

Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter_第1张图片

因此需要安装pytorch 1.4.0版本,cuda 为10.1版本的,其他的版本不支持。

接下来输入如下的命令安装:

pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  

pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-geometric

${CUDA}换为cu101,可成功安装

Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter_第2张图片

Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter_第3张图片

Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter_第4张图片

 

 

你可能感兴趣的:(跨平台编程)