torch安装包指定torch和cuda版本

torch安装包指定torch和cuda版本 1. 版本目录

版本库的列表: https://pytorch-geometric.com/whl/

pip install torch_sparse -f https://pytorch-geometric.com/whl/torch-1.9.1%2Bcu102.html

2. 选择合适的版本点击,生成对应的f参数链接

例如选择torch 1.9.1 cuda的版本10.2