I am using python 3.6.5 on Ubuntu 14.04.
I upgraded pytorch 0.4.0 via  ‘pip uninstall torch torchvision; pip install torch torchvision’
After upgrade, cuda operation for tensor or model gives me cuda runtime error(30).
Following is the example code.
device = torch.device(‘cuda’)
aaa = torch.Tensor(5).to(device)
or
aaa = torch.Tensor(5).cuda()
And here is the error message
Traceback (most recent call last):
File “”, line 1, in 
File “/home/kenkim/anaconda3/lib/python3.6/site-packages/torch/cuda/init.py”, line 161, in _lazy_init
torch._C._cuda_init()
RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:844
Could you tell me any fix that I should try?
p.s. When I downgrade to 0.3.0.post4, aaa.cuda() works without any error.
p.s. nvidia-smi output
±----------------------------------------------------------------------------+
| NVIDIA-SMI 384.111                Driver Version: 384.111                   |
|-------------------------------±---------------------±---------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  TITAN X (Pascal)    Off  | 00000000:01:00.0 Off |                  N/A |
| 23%   39C    P0    54W / 250W |      0MiB / 12189MiB |      0%      Default |
±------------------------------±---------------------±---------------------+
|   1  TITAN X (Pascal)    Off  | 00000000:02:00.0 Off |                  N/A |
|  0%   34C    P0    49W / 250W |      0MiB / 12189MiB |      0%      Default |
±------------------------------±---------------------±---------------------+
±----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
±----------------------------------------------------------------------------+