title: Pytorch cuda get_device_name and current_device() hang and are killed?
body: <p>I've just installed a new GPU (RTX 2070) in my machine alongside the old GPU. I wanted to see if PyTorch picked up it, so following the instructions here: <a href="https://stackoverflow.com/questions/48152674/how-to-check-if-pytorch-is-using-the-gpu">How to check if pytorch is using the GPU?</a>, I ran the following commands (Python3.6.9, Linux Mint Tricia 19.3)</p>
  
  <pre><code>&gt;&gt;&gt; import torch
  &gt;&gt;&gt; torch.cuda.is_available()
  True
  &gt;&gt;&gt; torch.cuda.current_device()
  Killed
  &gt;&gt;&gt; torch.cuda.get_device_name(0)
  Killed
  </code></pre>
  
  <p>Both of the two Killed processes took some time and one of them froze the machine for half a minute or so. Does anyone have any experience with this? Are there some setup steps I'm missing?</p>