linux 中手动安装 torch 的 cuda 版本,可以在如下官方地址中找到需要安装的版本

https://download.pytorch.org/whl/torch/

认准官网地址,其他的都不对!!!

复制链接,在需要安装的地方下载

wget https://download.pytorch.org/whl/cu121/torch-2.4.1%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=9a5f0b103cfe840b3568416aa5067f6e7b9fec67d9c5659fd43b1207450fe975

最后安装即可

pip install xxx.whl

补充链接,torch-scatter / torch-sparse 等工具库

https://pytorch-geometric.com/whl/

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐