跑深度学习模型Ⅱ:一文安装正确pytorch及dgl
*wait for? 2024-08-03 16:05:16 阅读 51
如果要用到GPU ,先看这篇安装并保证cuda可用。跑深度学习模型I:一文正确使用CUDA-CSDN博客
啊每次都是pytorch版本问题引发的一系列错误!
1. 查看自己的cuda版本
nvcc --version 可以看到我的cuda版本是11.8
nvidia-smi 查看NVIDIA驱动版本,可以看到我的驱动版本是522.06
方法二: 去NVIDIA控制面板,查看本机GPU支持的CUDA版本
具体操作:NVADIA的控制面板>帮助>系统信息>组件,可以看到我的版本是CUDA11,选择对应的版本。
我的GPU支持11.8的CUDA版本
2. 安装Pytorch包
在Pytorch官网中PyTorch,根据自己的CUDA版本选择,复制安装命令
不过要注意支持的CUDA版本。同时注意python版本支持情况,可查看这个表确认PyTorch中torch、torchvision、torchaudio、torchtext版本对应关系_torch2.0.1对应的torchvision-CSDN博客
我的python是3.9 ,为了后续能够安装我cuda版本能装的dgl,这里选择了2.2的torch
在pytorch官网中找到这个命令
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
把刚刚复制的安装命令,切换到需要用的python虚拟环境中执行。
环境创建切换等基础可以回看这篇【Python环境配置基础】-CSDN博客
检查torch库是否安装正确
import torch
print(torch.__version__)
检查cuda是否可用
torch.cuda.is_available()
终于可用啦啊啊啊啊太不容易了
其他torch相关库跑深度学习模型Ⅲ:正确安装与torch版本对应的其他torch包-CSDN博客
3. 安装DGL
进入DGL官网
根据已安装的Pytorch版本和cuda版本选择对应的命令,所以前两步版本都要安装正确才行!
【注意 dgl对于windows系统似乎只提供2.2.1版本的了,所以我好像试了很多次,最后前面装toroch时就安装2.2.1版本才行】
血与泪的教训!朋友们,一定要用conda装!
若想删除已经安装的不需要的DGL版本,则删除现有版本:
默认删除当前DGL:pip uninstall dgl
(cuda11.8可根据自己的环境调整)
再重新安装
4. 其他pytorch 库
以下连接可找其他库,如torch_scatter,torch_sparse等
pytorch-geometric.com/whl/torch-1.10.2%2Bcu113.html
5. 最后在安装torch时出现以下报错解决
pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='download.pytorch.org', port=443): Read timed out.
只需要使用清华镜像源 -ihl/cu118/pyphttps://pypi.tuna.tsinghua.ed
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。