cuda12.2安装DGL版本
时间: 2023-11-08 17:05:10 浏览: 1121
DGL库CPU版本和CUDA版本
安装 DGL 时需要先安装 PyTorch 或 TensorFlow,然后再通过 pip 安装 DGL。在安装 PyTorch 或 TensorFlow 时需要根据 CUDA 版本选择对应的版本。因此,安装 DGL 时需要先确定 CUDA 版本,然后选择对应的 PyTorch 或 TensorFlow 版本进行安装。
以 CUDA 12.2 为例,可以通过以下命令安装 PyTorch 和 DGL:
```
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
pip install dgl-cu111
```
其中,`torch==1.9.0+cu111` 表示安装 PyTorch 1.9.0 版本,使用的 CUDA 版本为 11.1。`dgl-cu111` 表示安装 DGL 的 CUDA 11.1 版本。
阅读全文