作者:marsggbo
链接:https://zhuanlan.zhihu.com/p/93278639
来源:知乎
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
在安装使用detectron2的时候碰到Kernel not compiled with GPU support 问题,前后拖了好久都没解决,现总结一下以备以后查阅。
不想看心路历程的可以直接跳到最后一小节,哈哈哈。
environment
因为我使用的是实验室的服务器,所以很多东西没法改,我的cuda环境如下:
- ubuntu
nvcc
默认版本是9.2nvidia-smi
版本又是10.0的
我之前一直没搞清楚这nvcc
和nvidia-smi
版本为什么可以不一样,想了解原因的可以看一下我之前的文章显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?。
reproduce
我一般都用Anaconda来安装pytorch,第一次安装的时候使用的如下命令安装的:
conda create -n myenv python=3.7 conda activate myenv conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
按理说这个命令会给myenv
环境安装cuda编译器和驱动等,但是在运行代码的时候还是会出现标题中的报错信息。我猜可能是因为detectron2在build的时候使用的是/usr/local
路径下的cuda compiler