@Royi I usually do the following, from bash:
python -c 'import torch; print(torch.rand(2,3).cuda())'
If the first fails, your drivers have some issue, or you dont have an (NVIDIA) GPU
If the second fails, your pytorch instalaltion isnt able to contact the gpu for some reason (eg you didnt do
conda install cuda80 -c soumith etc…)
(edit: if both the above succeed; I never saw any configuration error beyond that, other than my own coding error BUT if you try to run on a V100, using cuda 8 pytorch, the second statement will hang for ~5 minutes, whilst it creates the cache. But it’ll do this each time, so it’s useless, and you’ll need to use cuda 9 pytorch, (or not use a V100))