版权声明:欢迎转载。转载请注明地址:https://blog.csdn.net/weixin_32820767 https://blog.csdn.net/weixin_32820767/article/details/81632531
在docker 容器内运行 caffe2的时候。报错说
error == cudaSuccess. 30 vs 0. Error at: /packages/caffe2/caffe2/core/common_gpu.cc:117
报错信息没有及时记录,我的解决方法是:
docker run --runtime=nvidia ...
要么
nvidia-docker run ....