dgl的cuda版本安装+对应torch的cuda版本安装

超、凢脫俗 2024-02-18 12:48 98阅读 0赞

### 优秀教程 ###

1、[pytorch 1.8.1 + CUDA11.1 对应的DGL版本是0.6.1（linux、windows也适用）][pytorch 1.8.1 _ CUDA11.1 _DGL_0.6.1_linux_windows]  
2、[CUDA+pytorch+DGL安装][CUDA_pytorch_DGL]  
3、[安装dgl-cuda][dgl-cuda]  
4、[torch 安装备忘录][torch]

### dgl文件下载 ###

https://anaconda.org/dglteam/repo

![在这里插入图片描述][2d3e71a19ab94f7e84c37ad25f5ed34e.png]

可以将下面网址中的`cu111`改成你想要的版本，比如`cu113`、`cu117`等

https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu111/

![在这里插入图片描述][fedc8d6738fa43b7827200bfb1812405.png]

### pytorch文件下载 ###

https://download.pytorch.org/whl/torch_stable.html

![在这里插入图片描述][975f32f40a9d4dc6a147c92613c21466.png]

### PyTorch中torch、torchvision、torchaudio版本对应关系 ###

https://blog.csdn.net/AdamCY888/article/details/131021747
    
    https://blog.csdn.net/shiwanghualuo/article/details/122860521

> pytorch 1.8.1 + CUDA11.1 对应的DGL-cu111版本是0.6.1

![在这里插入图片描述][6e7c624ee27b4aa58b0a1b4c9a023f33.png]  
![在这里插入图片描述][88660870042e4774886056fbde8d3038.png]

### 我的环境 ###

显卡: RTX 3060
    cuda: cu111
    python: 3.7
    dgl-cu111: 0.6.1
    torch: 1.8.1+cu111

### 安装结果 ###

![在这里插入图片描述][48ad41237e974e4794974c3c085c7755.png]

![在这里插入图片描述][89ee0e1e2cbe49988e8a0c8ba9b9a64c.png]

### 调用程序 ###

![在这里插入图片描述][3ef4b83c8ee540af85a0ce5ee5fff2da.png]

[pytorch 1.8.1 _ CUDA11.1 _DGL_0.6.1_linux_windows]: https://blog.csdn.net/jiyunjiyun/article/details/132392632
[CUDA_pytorch_DGL]: https://blog.csdn.net/qq_39542531/article/details/124211882
[dgl-cuda]: https://blog.csdn.net/mrslw/article/details/130383993
[torch]: https://blog.csdn.net/AdamCY888/article/details/131021747
[2d3e71a19ab94f7e84c37ad25f5ed34e.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/be06cdd8525843fc8720cd3283e10e7b.png
[fedc8d6738fa43b7827200bfb1812405.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/4521b2a82ad44c9db39e4b2be8620f13.png
[975f32f40a9d4dc6a147c92613c21466.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/d3bc9849e42b4c38b7eabd5466325cfa.png
[6e7c624ee27b4aa58b0a1b4c9a023f33.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/6b513f2881ee41e3a114907294da94d3.png
[88660870042e4774886056fbde8d3038.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/b7ace71c01124541be7df4495fee8f34.png
[48ad41237e974e4794974c3c085c7755.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/1ac09941045d456195542db0006d6db3.png
[89ee0e1e2cbe49988e8a0c8ba9b9a64c.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/b7e5680dfa144681a90c2f788938f658.png
[3ef4b83c8ee540af85a0ce5ee5fff2da.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/c4b0d4d979c34e00a7feadae885f905b.png