docker task5

天池龙珠计划Docker训练营

没用GPU的报错如下：

---------------------------------------------------------------------------

RuntimeError                              Traceback (most recent call last)

<ipython-input-3-69f55cbb8b56> in <module>

     4 a = torch.randn(3,3)

     5 b = torch.randn(3,3)

----> 6a = a.to(device)

     7 b = b.to(device)

     8 c = torch.matmul(a,b)

/opt/conda/lib/python3.6/site-packages/torch/cuda/__init__.py in _lazy_init()

   170         # This function throws if there's a driver initialization error, no GPUs

   171         # are found or any other error occurs

--> 172         torch._C._cuda_init()

   173         # Some of the queued calls may reentrantly call _lazy_init();

   174         # we need to just return without initializing in that case.

RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx