got Unexpected error from cudaGetDeviceCount() after calling torch.cuda.is_available() about pytorch HOT 3 CLOSED

> nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Feb__7_19:32:13_PST_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0

#test.cu
#include <stdio.h>

__global__ void add(int *a, int *b, int *c) {
    c[blockIdx.x] = a[blockIdx.x] + b[blockIdx.x];
}

int main() {
    int a[5] = {1, 2, 3, 4, 5};
    int b[5] = {10, 20, 30, 40, 50};
    int c[5] = {0};

    int *d_a, *d_b, *d_c;
    int size = sizeof(int) * 5;

    cudaMalloc((void **)&d_a, size);
    cudaMalloc((void **)&d_b, size);
    cudaMalloc((void **)&d_c, size);

    cudaMemcpy(d_a, a, size, cudaMemcpyHostToDevice);
    cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice);

    add<<<5,1>>>(d_a, d_b, d_c);

    cudaMemcpy(c, d_c, size, cudaMemcpyDeviceToHost);

    printf("结果：");
    for (int i = 0; i < 5; i++)
        printf("%d ", c[i]);
    printf("\n");

    cudaFree(d_a);
    cudaFree(d_b);
    cudaFree(d_c);

    return 0;
}

> nvcc test.cu -o test
> ./test 
结果：0 0 0 0 0 

os.environ['CUDA_VISIBLE_DEVICES'] = "0"

lspci -v
63b6:00:00.0 3D controller: Microsoft Corporation Device 008e
	Physical Slot: 3509819646
	Flags: bus master, fast devsel, latency 0, NUMA node 0
	Capabilities: <access denied>
	Kernel driver in use: dxgkrnl
	Kernel modules: dxgkrnl

b330:00:00.0 3D controller: Microsoft Corporation Device 008e
	Physical Slot: 3646467553
	Flags: bus master, fast devsel, latency 0, NUMA node 0
	Capabilities: <access denied>
	Kernel driver in use: dxgkrnl
	Kernel modules: dxgkrnl