我在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
。
这真的是问题吗?如何更改系统上的配置和环境上的配置以使其匹配?