pytorch官网：https://pytorch.org/ 进入官网后往下拉，在下面这个界面可以下载最新版，compute platform中...
你的意思是在下面这个地方选择CUDA的版本，这个地方的版本就是对应着gpu版的pytorch。
kmeans_pytorch是支持GPU的pytroch写的kmeans包,但官网文档叙述不清晰,重新加了点注释 - lvszl/kmeans_pytorch_new