Scene-Graph-Benchmark.pytorch(KaihuaTang)代码配置过程记录

本文记录了在服务器上安装CUDA11.0,并配置Scene-Graph-Benchmark.pytorch项目的详细步骤,包括conda环境安装PyTorch1.7.1,解决gcc错误,设置CUDA_HOME环境变量,以及处理python setup.py安装中的问题。
摘要由CSDN通过智能技术生成
  1. GitHub地址:https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch

  2. 我的配置:
    在这里插入图片描述

  3. 原服务器中已经装了:cuda10.1和cuda9.0,但是想要配置cuda11.0。

  4. 配置过程

  • 第一步是在服务器上安装cuda11.0.参照:
    https://zhuanlan.zhihu.com/p/122286055
  • 第二步:根据https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/master/INSTALL.md,其中在
    conda install pytorch1.4.0 torchvision0.5.0 cudatoolkit=10.1 -c pytorch 更改为
    conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
    (其他版本可以根据https://pytorch.org/get-started/previous-versions/进行选择)
  • 其它步骤与https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/master/INSTALL.md相同。
  1. 可能遇到的问题:
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值