在系统上配置了CUDA,在工作环境上配置了CUDA

我在Google colab上从https://github.com/jwyang/faster-rcnn.pytorch运行了更快的rcnn算法。

我正在使用torch == 0.4.1,torchfile == 0.1.0和torchvision == 0.2.0。

当我运行trainval.py脚本时,出现此错误:

  

ImportError:   /content/faster-rcnn.pytorch/lib/model/roi_crop/_ext/roi_crop/_roi_crop.so:   未定义的符号:__cudaPopCallConfiguration

我认为问题是我的系统上可能配置了CUDA 10.0,并且在工作环境中使用的是CUDA 10.1。

通过nvidia-smi命令,我得到:CUDA Version: 10.1

通过nvcc --version命令,我得到:Cuda compilation tools,release 10.0,V10.0.130

这真的是问题吗?如何更改系统上的配置和环境上的配置以使其匹配?

eywe5275 回答:在系统上配置了CUDA,在工作环境上配置了CUDA

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/2892714.html

大家都在问