pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html pip install torch-geometric
${CUDA}
and ${TORCH}
should be replaced by your specific CUDA version (cpu
, cu92
, cu101
, cu102
) and PyTorch version (1.4.0
, 1.5.0
, 1.6.0
), respectively.
版权声明:本文为Sinsinw原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。