cuda11对应的torch_scatter是什么
时间: 2024-04-03 18:34:34 浏览: 95
torch_scatter-2.0.7-cp36-cp36m-linux_x86_64whl.zip
如果你正在使用 CUDA 11,可以使用以下命令安装与之相匹配的 `torch_scatter`:
```
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0%2Bcu111.html
```
这个命令会安装 PyTorch 1.9.0 和 CUDA 11 相匹配的 `torch_scatter` 版本。注意,这个命令需要联网下载安装包,所以你需要保证远程机器可以访问互联网。如果你的网络环境无法直接下载安装包,可以先在本地机器上下载安装包,然后通过 `scp` 命令将安装包传输到远程机器上进行安装。
阅读全文