Ubuntu配置Pytorch on Graph (PoG)环境过程图解

最近准备复现一下 KDD-20 Towards Deeper Graph Neural Networks 的代码，顺便学习一下 GCN 最新的一些进展。

GCN 的代码通常需要安装 Pytorch on Graph, 按照其 官方指导, 发现无法兼容当前所使用的服务器上的 pytorch1.3 和 cuda10.0 环境, 需要升级 cuda 和 pytorch 的版本。

于是准备重新安装一个虚拟环境，使用上 cuda 10.2 和 pytorch1.6。

1. 安装 Cuda10.2 以及对应版本的 Cudnn

安装 cuda 10.2，

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

然后安装对应版本的 cudnn

但由于当前机器上已经安装了 cuda 10.0，因此想要找一些能够切换的方法，不删去当前的 cuda10.0, 发现可以使用软链接的方式进行切换

# 切换为 cuda 8.0 版本
rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda/
nvcc --version #查看当前 cuda 版本

 nvcc: NVIDIA (R) Cuda compiler driver
 Copyright (c) 2005-2016 NVIDIA Corporation
 Built on Mon_Jan_23_12:24:11_CST_2017
 Cuda compilation tools, release 8.0, V8.0.62

# cuda8.0 切换到 cuda9.0
rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0/ /usr/local/cuda/
nvcc --version

安装 cuda10.2，这里只安装 Cuda Toolkit 10.2, 不安装 driver, 因为之前 Cuda10.0 已经装过 driver 了。

然后尝试安装 对应版本的 cudnn, 在linux上

使用

wget "https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.3.33/10.2_20200825/cudnn-10.2-linux-x64-v8.0.3.33.tgz"

会导致一直显示 403 Forbid 错误，但用 windows 浏览器可以访问成功，用 windows 下载，但后缀名变为了 ".solitairetheme8",

mv cudnn-10.2-linux-x64-v8.0.3.33.solitairetheme8 cudnn-10.2-linux-x64-v8.0.3.33.tgz

接下来安装 cudnn，注意到，现在有多个版本的 cuda 了，不要再使用软链接 /usr/local/cuda/， 而应该使用对应版本的 cuda 地址。

使用 cd /usr/local/ 可以看到当前安装的 cuda 的情况， 我这里是：

使用如下命令完成 cudnn 的安装：

tar -zxvf cudnn-10.2-linux-x64-v8.0.3.33.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

从官网下载实在太慢，也可以改为设置为 清华源

# 首先执行如下几条命令更换清华镜像源conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

安装结果：

突然发现，conda 也会安装一个 cudatoolkit=10.2, 但这个和自己手动安装的还是有很大的不同的

安装 pytorch1.7 一直没有成功，最新版的 pytorch1.7 需要从国外官网源头直接进行下载，速度太慢，一直失败。
所以尝试安装 pytroch 1.5，

conda install pytorch=1.5

conda install torchvision

然后再按照 PoG 官网安装相应的包：

pip install torch-scatter
pip install torch-sparse
pip install torch-cluster
pip install torch-spline-conv
pip install torch-geometric

最后终于成功完成了安装。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。

赞 (0)

相关推荐

  • Ubuntu 20.04系统中安装vncserver的方法步骤

    Ubuntu 20.04已经在2020年4月份正式发布,笔者今天正式安装了一个镜像系统,并在该系统中安装vncserver以便于远程图形化管理和使用.安装vncserver的步骤如下: 第一步:安装桌面环境和VNCServer #更新 apt update #安装桌面环境 sudo apt install xfce4 xfce4-goodies #安装vncserver apt install tightvncserver #启动vncserver vncserver 随后按照提示,输入vncs

  • Ubuntu修改密码及密码复杂度策略设置方法

    一.修改密码 1.修改普通用户密码 passwd 先输入当前密码确认,然后输入新的密码修改 2.修改root用户密码 sudo passwd root 默认root用户被禁止登录,如果需要解除限制,修改配置即可 sudo vim /etc/ssh/sshd_config 将默认配置注释掉,添加一行新的配置,默认的配置为允许root登录,但是禁止root用密码登录 PermitRootLogin prohibit-password → PermitRootLogin yes sudo servic

  • Ubuntu20.04 VNC 安装与设置实现

    VNC是一个远程桌面协议.按照本文的说明进行操作可以实现用VNC对Ubuntu 20.04进行远程控制.一般的VNC安装方式在主机没有插显示器的时候是无法使用的.下面的操作可以在主机有显示器和没有显示器时都能够正常工作. 首先安装x11vnc sudo apt-get install x11vnc -y 设置vnc密码 sudo x11vnc -storepasswd /etc/x11vnc.pass 创建x11vnc自启动服务 创建 /etc/systemd/system/x11vnc.ser

  • ubuntu下 AndroidStudio4.1启动报错问题的解决

    报错信息如下: Internal error. Please refer to https://code.google.com/p/android/issues java.lang.NoSuchMethodError: com.intellij.ide.plugins.PluginManagerCore.loadDescriptors()[Lcom/intellij/ide/plugins/IdeaPluginDescriptorImpl; at com.a.a.b.b.ar.a(ar.java

  • Ubuntu20.04安装配置GitLab的方法步骤

    介绍 GitLab CE或Community Edition是一个开源应用程序,主要用于托管Git存储库,以及其他与开发相关的功能,如问题跟踪.它旨在使用您自己的基础架构进行托管,并为您的开发团队提供部署内部存储库的灵活性,与用户交互的公共方式,或者为贡献者提供托管自己项目的方法. GitLab项目使用简单的安装机制在您自己的硬件上设置GitLab实例变得相对简单.在本指南中,我们将介绍如何在阿里云Ubuntu 20.04服务器上安装和配置GitLab. 前提条件 部署GitLab的实例要求至少

  • ubuntu开启22端口的实现

    场景 需要通过xshell工具,通过ssh连接ubuntu服务器.默认端口22. 结果 常规连接 命令行连接 //直接通过命令连接 ssh 远程机器名@192.168.67.128 或者 telnet 192.168.67.128 22 分析 ip和端口都没有问题. 查看服务器22端口是否启动. lsof -i:22 //安装后需要重启 sudo apt-get install openssh-server openssh-client service ssh start ssh localho

  • Vscode远程连接Ubuntu出错问题的解决方法

    一.事件背景: 由于工作需要使用vscode远程连接到Ubuntu.(具体连接方法可百度,本文仅限分享其中遇到的问题) 二.问题现象: 在这过程中遇到了一个问题.其文字描述为:Could not establish connection to "IP地址"与过程试图写入管道不存在. 如下图所示: 三.问题分析: 1.根据问题报错在网上查找原因. 根据关键字"过程试图写入的管道不存在"匹配到的东西太多太广了,例如由于安装Git导致,由于Ubuntu升级系统导致,由于未

  • Ubuntu中使用VS Code与安装C/C++插件的教程详解

    优麒麟Ubuntu20.04中使用VS Code. VS Code的版本是1.48.0. 以下内容仅限于上述环境,对于Windows环境下的使用虽然类似,只能参考使用. 1.准备 a.软件安装 可以通过软件商店进行安装,也可以下载安装. 下载安装的官网地址:https://code.visualstudio.com/. 可以下载deb和rpm两种格式的安装软件. b.目录计划 编写程序过程中,需要事先决定编写的程序放在哪里,这就需要做一个目录计划.任何一个软件项目也都需要做目录规划,以便所写的程

  • Ubuntu配置Pytorch on Graph (PoG)环境过程图解

    最近准备复现一下 KDD-20 Towards Deeper Graph Neural Networks 的代码,顺便学习一下 GCN 最新的一些进展. GCN 的代码通常需要安装 Pytorch on Graph, 按照其 官方指导, 发现无法兼容当前所使用的服务器上的 pytorch1.3 和 cuda10.0 环境, 需要升级 cuda 和 pytorch 的版本. 于是准备重新安装一个虚拟环境,使用上 cuda 10.2 和 pytorch1.6. 1. 安装 Cuda10.2 以及对应

  • 在Mac中PyCharm配置python Anaconda环境过程图解

    在 python 开发中我们最常用的IDE就是PyCharm,有关PyCharm的优点这里就不在赘述.在项目开发中我们经常用到许多第三方库,用的最多的命令就是pip install 第三方库名 进行安装.现在你可以使用一个工具来帮你解决经常安装第三方库的麻烦,这个工具就是Anaconda. Anaconda集成了100多个常用的第三方库,在项目的开发中就可以减少使用 pip 命令进行安装.有关Anaconda的安装包,大家可以前往官网进行下载并安装. 这里我想注重说明的是Mac 中PyCharm

  • Eclipse配置python开发环境过程图解

    1.下载python 安装包.python-2.6.6.msi.并安装. 默认python会安装在C:\Python26下,查看环境变量,如果没有在path路径中写入则手动添加.打开一个dos窗口,验证python是否安装成功: C:\Documents and Settings\Administrator>python Python 2.6.6 (r266:84297, Aug 24 2010, 18:46:32) [MSC v.1500 32 bit (Intel)] on win32 Ty

  • Pycharm配置lua编译环境过程图解

    一.Pycharm安装lua插件 pycharm-->setttings-->plugins--->lua ###安装lua插件,安装完毕重启pycharm 二.安装lua环境 https://github.com/rjpcomputing/luaforwindows/releases 注意:请根据你需要的lua版本下载对应的 安装完成,在cmd下输入lua测试是否安装成功 三.配置Pycharm lua开发环境新建luatest文件,Pycharm打开文件.新建lua脚本,输入prin

  • 基于Jenkins搭建.NET Core持续集成环境过程图解

    我们用NuGet还原.NET Core项目会报以下错误: error NETSDK1064: 未找到版本为 1.8.2 的包 BouncyCastle.NetCore.它可能已在 NuGet 还原后删除.否则,NuGet 还原可能只是部分完成,这种情况可能是最大路径长度限制所导致. 关于这个问题我找了好久 也是google到一条信息才明白 这里就直接放解决办法,其实网上就有方法,有的简短,有的笼统 我们这里用到的是dotnet命令来还原.构建.以及部署 从 .NET Core 2.0 开始,无需

  • Ubuntu配置sublime text 3的c编译环境的具体步骤

    Ubuntu配置sublime text 3的c编译环境的具体步骤

    c语言 选择tool –> Build System –> New Build System 然后输入以下代码 { "shell_cmd": "gcc \"${file}\" -o \"${file_path}/${file_base_name}\" -Wformat=2", "file_regex": "^(..[^:]*):([0-9]+):?([0-9]+)?:? (.*)$&q

  • VSCode IDE 配置环境过程解析

    目录 说明 前提条件 创建项目 导出模板项目 VSCode创建项目 配置C/CPP环境 配置编译任务 如果用 PlatformIO 创建 libopencm3 项目可以做到零配置, 只是 libopencm3 的版本会旧一点. 这里说的是仅使用 VSCode 创建C/CPP项目时的配置. VSCode 有代码提示, 定位来源和各种快捷键, 更适合日常编码工作. 说明 因为 PlatformIO 的 platform:st-stm32 自带 libopencm3, 如果用 PlatformIO 创

  • vuecli3.0脚手架搭建及不同的打包环境配置vue.config.js的详细过程

    参考文档 https://cli.vuejs.org/zh/ 1.安装 npm install -g @vue/cli 2.检查安装 vue -V 3.创建项目 vue create project_name 出现下图信息 4.上图两种选择方式,第一种是默认的,第二种自己选择配置  这里一般选择第二种自己配置,点击回车键后出现下图信息 5.在上图中选择你需要的模块,上下移动,空格进行确定,下图是测试选择的 6.选好后,敲Enter键,接着选择 7.选择完毕后,一路Enter,到下图 8.模板创建

  • Ubuntu服务器下搭建php运行环境的方法

    本文实例讲述了Ubuntu服务器下搭建php运行环境的方法.分享给大家供大家参考,具体如下: 安装 Apache2: sudo apt-get install apache2 安装PHP模块: sudo apt-get install php5 安装Mysql sudo apt-get installmysql-server 其他模块安装: sudo apt-get install libapache2-mod-php5 sudo apt-get installlibapache2-mod-au

  • win10从零安装配置pytorch全过程图文详解

    1.安装anaconda (anaconda内置python在内的许多package,所以不用另外下载python) 可以点击下面的清华开源软件镜像站,在官网下载anaconda不如在这下的快 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 镜像站 我下载的是下图这个版本,对应的是3.7,anaconda3代表对应python3 下载过程中基本上都是默认选项 下面这步可以不勾选add anaconda to my PATH envir

随机推荐