-
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相同。
- 可能遇到的问题: