深度学习环境配置避坑-CUDA11.0+DGL1.8

深度学习环境配置避坑-CUDA11.0+DGL1.8

参考教程

之前已经写过详细的CUDA+Pytorch+DGL安装博文了,Linux系统可参考这里,Windows系统可参考这里。本篇博文仅用于记录一些环境配置中可能遇到的坑,及避坑指南(本人亲亲测有效)

坑1:CUDA版本与GPU型号不符

报错:GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
原因:GeForce RTX 3080算力为8.6,仅适合CUDA11.0或CUDA11.1,安装其他版本CUDA则会报错
避坑指南:此时如果按照Pytorch官网,用命令行

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

安装Pytorch,则自动安装CPU版本Pytorch,导致后续实验测试一直报错。经过一下午的努力,终于找到一篇成功的安装方法,用命令行

pip install torch=1.8.0 torchvision=0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

安装成功。

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装DGLCUDA版本可以按照以下步骤进行操作: 1. 首先,进入DGL官网,在根据你的CUDA版本选择相应的命令。你可以复制"Run this"后面的命令,例如:`conda install -c dglteam dgl-cuda10.2`。 2. 打开终端,将复制的命令粘贴到终端中,然后按下回车键。等待终端出现是否同意继续的提示,输入"y"来继续安装。 3. 如果你想指定安装特定版本的DGL,首先需要删除当前已安装的版本。你可以使用以下命令删除当前版本:`conda uninstall -c dglteam dgl-cuda10.2`(cuda10.2根据你的环境进行调整)。如果你想要删除指定版本,可以使用以下命令:`conda uninstall -c dglteam dgl-cuda10.2==0.5.0`(cuda10.2==0.5.0根据你的环境进行调整,你可以使用`conda list`命令查看当前使用的版本)。 4. 如果你还没有安装DGL,你可以直接指定要安装的版本。例如:`conda install -c dglteam dgl-cuda10.2==0.4.3post2`。你可以在DGL的官网上查看版本名称的网站。 总结:要安装DGLCUDA版本,你可以根据你的CUDA版本选择相应的命令,并在终端中执行。如果需要指定特定版本,可以先删除当前已安装的版本,然后安装指定版本。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [pytorch系列(三)——cudaDGL库安装](https://blog.csdn.net/qq_30049011/article/details/120763171)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [深度学习环境配置避坑-CUDA11.0+DGL1.8](https://blog.csdn.net/weixin_46065198/article/details/125642546)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值