-
GitHub地址:https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch
-
我的配置:

-
原服务器中已经装了:cuda10.1和cuda9.0,但是想要配置cuda11.0。
-
配置过程
- 第一步是在服务器上安装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相同。
- 可能遇到的问题:
-
cannot create temp file for here-document: No space left on devi

本文记录了在服务器上安装CUDA11.0,并配置Scene-Graph-Benchmark.pytorch项目的详细步骤,包括conda环境安装PyTorch1.7.1,解决gcc错误,设置CUDA_HOME环境变量,以及处理python setup.py安装中的问题。
最低0.47元/天 解锁文章
2453

被折叠的 条评论
为什么被折叠?



