多模态第2篇:MMGCN代码配置

一、Windows环境

1.创建并激活虚拟环境

#创建虚拟环境命名为mmgcn,指定python版本为3.8
conda create -n mmgcn python=3.8
#激活虚拟环境
conda activate mmgcn

2.安装pytorch

#torch2.0.0 + cu118
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

3.安装torch_geometric

torch_geometric有四个依赖库,torch-scatter、torch-sparse、torch-cluster、torch-spline-conv,直接 pip安装不行,依赖库也要一起安装,命令如下:

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-2.0.0+cu118	.html

报错:在执行过程中出现安装包卡住不动,无法继续安装成功的问题,报错如下Building wheel for torch-scatter (setup.py) …
原因:相关的wheel文件下载不到 或者是下载缓慢;
解决:在网站中下载自己对应版本的whl文件:下载地址
首先选择对应的torch版本和cuda版本:
在这里插入图片描述
然后下载四个依赖库:
在这里插入图片描述
下载完毕后,cd进入到存放的文件夹,执行pip命令进行安装whl文件:
在这里插入图片描述

conda activate mmgcn
cd C:\Users\孟静\Desktop\torch2.0.0+cu118 torch_geometric
pip install torch_cluster-1.6.1+pt20cu118-cp38-cp38-win_amd64.whl
pip install torch_scatter-2.1.1+pt20cu118-cp38-cp38-win_amd64.whl
pip install torch_sparse-0.6.17+pt20cu118-cp38-cp38-win_amd64.whl
pip install torch_spline_conv-1.2.2+pt20cu118-cp38-cp38-win_amd64.whl
#最后再安装torch_geometric
pip install torch_geometric

4.根据代码标红,又安装了:

pip install pandas
pip install tensorboardX

5.修改train.py

train.py中第2行的CUDA_VISIBLE_DEVICES改为0
在这里插入图片描述

6.Windows系统在pycharm上运行.sh文件

(1)下载并安装Git,默认设置安装即可 下载链接 :Git下载
(2)在pycharm中进行设置

File—>Settings—>Tools—>Terminal—>Shell path

在这里插入图片描述

将Shell path一栏改成刚刚下载的git的路径,注意选择的是sh.exe,不是git.exe!

在这里插入图片描述
(3)最后在左下角打开Terminal,即可运行.sh文件了。

打开run.sh文件,直接点击左侧的绿色三角形即可开始运行。
在这里插入图片描述

二、Unbuntu系统

由于cuda出现问题导致训练特别慢,但是能够正常运行,报错如下:

Could not load symbol cublasGetSmCountTarget from libcublas.so.11. Error: /usr/local/cuda-11.3/lib64/libcublas.so.11: undefined symbol: cublasGetSmCountTarget

该问题还未解决…

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值