linux 下安装pythorch-gpu+dgl

原创 已于 2022-07-15 14:34:41 修改 · 2.1k 阅读 · 3 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#pytorch #深度学习 #人工智能

于 2022-07-15 14:18:04 首次发布

linux 下安装pythorch-gpu+cuda+cudnn+dgl

  • 确定cuda和cudnn的安装
  • 确定pytorch的gpu版本

一查找

查看当前显卡驱动支持的cuda

nvidia-smi

在这里插入图片描述

二安装

找到合适的cuda版本，在pytorch官网上找到pytorch对应的版本运行给出的安装命令。
在这里插入图片描述

 conda search cudatoolkit --info

找到和显卡支持的cuda对应的版本，红框标出的是此版本支持的cuda，实验用的这台服务器最大支持的是11.6的cuda，因此我们选用支持高于11的的cudatoolkit

在这里插入图片描述
安装命令是

conda install cudatoolkit
最低0.47元/天 解锁文章

200万优质内容无限畅学
在Conda中搭建PyTorch+DGL+PyG时需要仔细查看的几个关于版本的网页
qysh123的专栏
02-01 576
但是现在还是能遇到问题，例如今天在一个新的Ubuntu上复现别人论文里的实验，就还是出现了问题。没办法，还是得仔细检查版本。特别应该注意PyTorch和CUDA之类的对应关系。
Linux安装PyTorch和DGL的GPU版本
Wudaoyunqi的博客
03-12 1824
复现一篇论文，要求指定pytorch>=1.8.1，dgl==0.4.0（后续发现0.4.0~0.4.3的都可以，高版本不支持是因为API包变了，其实自己小小改一下代码高版本也能用，但是我懒不想折腾）搜了搜没啥有用的信息，我选择不折腾，最后安装的是pytorch = 1.10.1，dgl = cu102-0.4.2，完事了。，cu版本的只有90, 100，101，102, 110，111，挑一个最接近你cuda版本的（向下兼容），直接pip install xxx.whl。
Linux 安装 pytorch+cuda+gpu 大模型开发环境过程记录
最新发布
恒
05-18 525
在安装好 PyTorch 后，安装 Hugging Face 的 transformers 库（核心工具库）和其他相关依赖的步骤如下。因为系统已有 Anaconda3 和 minianaconda3，但是已有的这些环境存在问题。为不破坏这些环境，我重新安装一个 Anaconda3，与已有的共存，切记不可自动设置。查看系统 cuda 版本，安装与之匹配的 pytorch：2.6.0。指定安装目录：/llm/huggingface/anaconda3。开始你的大模型开发之旅吧！[no]不要配置: 直接回车。
pytorch 1.8.1 + CUDA11.1 对应的DGL版本是0.6.1（linux）
jiyunjiyun的博客
08-20 2316
pip install dgl-cu111==0.6.1
DGL 和 torch 对应关系
Python领域优质萌新学习笔记
08-18 9353
调整版本号下载dgl， 运行本地代码，若运行失败则删除dgl，并继续降低版本号安装，如此循环直至脚本运行成功。dgl 1.1.0 这个版本对应python39 的 torch 1.9。报错提示需要升级torch的版本，换言之需要降低dgl的版本。也可以对应 python37 的 torch 1.10.2。可以先查看本地dgl的版本。
torch、dgl、python、geometric、pycharm对应版本
weixin_74749569的博客
12-28 629
torch与dgl与python的版本对应问题
图神经网络环境配置：Win11+CUDA10.2+Pytorch1.10+DGL
weixin_46065198的博客
06-14 2800
过程参考我另一篇博文，配置好numpy等其他依赖包。 激活虚拟环境 查询pytorch和对应CUDA 进入Pytorch官网，根据需要的Pytorch和CUDA版本查看conda命令。 此处我配置Pytorch1.10+CUDA10.2，命令为： 运行太慢或网络连接失败，手动从以下4个网址查询需要的包，如果有则指定对应channel 这里使用以下命令运行成功，注意必须http开头才能成功。 虚拟环境下安装CUDA版本的DGL 打开DGL官网，直接开可能会很慢，搭梯子之后明显变快了。根据CUDA版本获得对应
torch_cluster-1.6.0+pt113cu117-cp39-cp39-win_amd64.whl
12-24
需要配和指定版本torch-1.13.0+cu117使用，请在安装该模块前提前安装官方命令安装torch-1.13.0+cu117对应cuda11.7和cudnn，注意电脑需要有nvidia显卡才行，支持GTX920以后显卡，比如RTX20 RTX30 RTX40系列显卡
【教程】Linux环境下支持GPU加速的Pytorch、PyG、DGL环境配置步骤记录
努力努力
01-12 2717
这篇步骤记录针对的是在Linux系统下配置支持GPU加速的Pytorch（其它系统的也可以参考基本步骤，但是涉及到的具体的选项和指令会有所不同），以及做图神经网络会涉及到的PyG、DGL库的安装过程。
torch_cluster-1.6.0+pt112cpu-cp39-cp39-linux_x86_64.whl.zip
12-13
以下是在没有GPU环境下安装`torch-1.12.0+cpu`的步骤： 1. 首先，确保Python环境为3.9版本，因为给定的`torch_cluster`版本是针对Python 3.9编译的。 2. 打开终端，输入以下命令安装`torch-1.12.0+cpu`： ``` pip...
浪潮V100环境配置pytorch环境和相应的DGL环境以及其他相关的库
qq_43678445的博客
07-04 653
因为V100的cuda版本为10.1所以下载pytorch版本为1.7。如果还要安装transformers可以安装4.1.1。先创建一个虚拟环境，安装python3.7版本。安装DGL只能匹配torch的版本为0.5.3。安装 scikit-learn。
DGL库CPU版本和CUDA版本
02-27
DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-win_amd64.whl win-64-dgl-cuda11.1-0.9.1-py37-0.tar.bz DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-win_amd64.whl win-64-dgl-cuda11.1-0.9.1-py37-0.tar.bz DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-win_amd64.whl win-64-dgl-cuda11.1-0.9.1-py37-0.tar.bz DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-win_amd64.whl win-64-dgl-cuda11.1-0.9.1-py37-0.tar.bz DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-win_amd64.whl win-64-dgl-cuda11.1-0.9.1-py37-0.tar.bz DGL库CPU版本和CUDA版本 dgl-0.9.1-cp37-cp37m-wi
dgl的cuda版本安装+对应torch的cuda版本安装
热门推荐
Python领域优质萌新学习笔记
11-20 2万+
pytorch 1.8.1 + CUDA11.1 对应的DGL-cu111版本是0.6.1。改成你想要的版本，比如。
跑深度学习模型Ⅱ：一文安装正确pytorch及dgl
qq_46147236的博客
08-02 1932
啊每次都是pytorch版本问题引发的一系列错误！nvcc --version 可以看到我的cuda版本是11.8nvidia-smi 查看NVIDIA驱动版本，可以看到我的驱动版本是522.06方法二： 去NVIDIA控制面板，查看本机GPU支持的CUDA版本具体操作：NVADIA的控制面板>帮助>系统信息>组件,可以看到我的版本是CUDA11,选择对应的版本。我的GPU支持11.8的CUDA版本在Pytorch官网中，根据自己的CUDA版本选择，复制安装命令。
安装指定版本的pytorch和DGL（GPU+Linux）
weixin_46266120的博客
04-25 1935
直接安装直接查询1.8.0。
超算云服务深度学习环境配置Pytorch1.6+CUDA10.2+DGL0.4.3
weixin_46065198的博客
05-18 7765
超算云服务深度学习环境配置登录超算云服务连接服务器1. 通过SSH连接服务器2. 通过第三方软件连接服务器（Xshell）配置深度学习环境1. 查看服务器已安装模块2. 调用Anaconda模块3. 创建Python3.7的虚拟环境4. 虚拟环境下安装CUDA10.2+Pytorch1.6.0
解决import dgl报错torch版本不匹配的问题
qq_41915671的博客
11-20 2346
跑一个gpu版本的代码需要用到dgl库，运行后报错RuntimeError: DGL requires PyTorch >= 1.9.0 ，是版本不匹配的问题，需要安装对应版本的dgl。哎 记录就花了几分钟 解决这个问题搞了我一个多小时 又是fw的一天。首先查看torch版本。
Linux（Ubuntu）配置Cuda，Pytorch，Anaconda
weixin_46568899的博客
10-28 2108
Linux（Ubuntu）配置Cuda，Pytorch，Anaconda
Python：23种Pandas核心操作方法，Python彻底组件化方案实践方法
weixin_58085973的博客
03-30 851
Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值