Error in using cuda

1

I can detect cuda but cannot use it .so weird

%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20180828221612
%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20180828221612960×720 113 KB

my configuration: 1080ti driver:390.77 cuda9.0 cudnn 7.0.5 py3.5
I am sure i correctily installed cuda cause “nvidia-smi” and “nvcc -V” presents normally ,and tensorflow-gpu works well

who can help me!

2

I usually use a.to(device) instead of .cuda() where the device is set like this:
device = torch.device(“cuda” if args.cuda else “cpu”)

3

yeah i also tried this and still the same error