Repository: pytorch
Issue: The cuda problem in caffe2
Labels: caffe2

Hi

I install Caffe2 with Python3.6.8, cuda 9.0.176, cudnn 7 ,Driver Version: 384.130 OS ubuntu 17.10

Caffe2 install use: conda install pytorch-nightly cuda90 -c pytorch

when I try to run maskrcnn demo as below the "CUDA driver version is insufficient for CUDA runtime version" error occurs . pls help me! Thanks a lot!
python tools/infer_simple.py \
    --cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml \
    --output-dir /tmp/detectron-visualizations \
    --image-ext jpg \
    --wts https://dl.fbaipublicfiles.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl \
    demo


THE ERROR DETAIL :

```
WARNING cnn.py:  25: [====DEPRECATE WARNING====]: you are creating an object from CNNModelHelper class which will be deprecated soon. Please use ModelHelper object with brew module. For more information, please refer to caffe2.ai and python/brew.py, python/brew_test.py for more information.
INFO net.py:  60: Loading weights from: /tmp/detectron-download-cache/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl
INFO net.py:  96: conv1_w loaded from weights file into gpu_0/conv1_w: (64, 3, 7, 7)
Traceback (most recent call last):
  File "/home/bruce/detectron/tools/infer_simple.py", line 185, in <module>
    main(args)
  File "/home/bruce/detectron/tools/infer_simple.py", line 135, in main
    model = infer_engine.initialize_model_from_cfg(args.weights)
  File "/home/bruce/detectron/detectron/core/test_engine.py", line 329, in initialize_model_from_cfg
    model, weights_file, gpu_id=gpu_id,
  File "/home/bruce/detectron/detectron/utils/net.py", line 112, in initialize_gpu_from_weights_file
    src_blobs[src_name].astype(np.float32, copy=False))
  File "/home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/workspace.py", line 352, in FeedBlob
    return _Workspace_feed_blob(ws, name, arr, device_option)
  File "/home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/workspace.py", line 711, in _Workspace_feed_blob
    return ws.create_blob(name).feed(arr, device_option)
  File "/home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/workspace.py", line 741, in _Blob_feed
    return blob._feed(arg, device_option)
RuntimeError: [enforce fail at common_gpu.cc:98] error == cudaSuccess. 35 vs 0. Error at: /opt/conda/conda-bld/pytorch-nightly_1560316055483/work/caffe2/core/common_gpu.cc:98: CUDA driver version is insufficient for CUDA runtime version
frame #0: c10::ThrowEnforceNotMet(char const*, int, char const*, std::string const&, void const*) + 0x59 (0x7f34ba1a08a9 in /home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/../../torch/lib/libc10.so)
frame #1: caffe2::CaffeCudaGetDevice() + 0x8f6 (0x7f346effedc6 in /home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/../../torch/lib/libcaffe2_gpu.so)
frame #2: <unknown function> + 0x2cf8165 (0x7f3470847165 in /home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/../../torch/lib/libcaffe2_gpu.so)
frame #3: <unknown function> + 0x68eb7 (0x7f34ba8c2eb7 in /home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/caffe2_pybind11_state_gpu.cpython-36m-x86_64-linux-gnu.so)
frame #4: <unknown function> + 0x6a1ab (0x7f34ba8c41ab in /home/bruce/anaconda3/envs/maskrcnn/lib/python3.6/site-packages/caffe2/python/caffe2_pybind11_state_gpu.cpython-36m-x86_64-linux-gnu.so)
```