Why pytorch only run on DEVICE:GPU0

1

When import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “1”

ERROR occured:RuntimeError: CUDNN_STATUS_ARCH_MISMATCH

2

What GPUs have you installed?
It must support compute capability >= 3.0.

3

GPU 0：memory:1G
GPU 1: memory:12G,compute capability = 3.5
why occured this question?Thanks very much!!!

4

Hmm, ok. Could you please post what nvidia-smi prints in the terminal?

5

捕获
捕获741×448 12.1 KB