Python 运行时错误：CUDA未知错误-这可能是由于环境设置不正确造成的，例如更改环境变量CUDA\u VISIBLE\u设备

以下是英伟达smi的输出

  (pytorch) kumar@kumar:~$ nvcc -V
    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2021 NVIDIA Corporation
    Built on Sun_Feb_14_21:12:58_PST_2021
    Cuda compilation tools, release 11.2, V11.2.152
    Build cuda_11.2.r11.2/compiler.29618528_0
    (pytorch) kumar@kumar:~$ 

然而，当我试着跑的时候

    Thu Apr  8 15:04:49 2021       
    +-----------------------------------------------------------------------------+
    | NVIDIA-SMI 460.67       Driver Version: 460.39       CUDA Version: 11.2     |
    |-------------------------------+----------------------+----------------------+
    | GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
    | Fan  Temp  Perf  Pwr: Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
    |                               |                      |               MIG M. |
    |===============================+======================+======================|
    |   0  GeForce RTX 3070    Off  | 00000000:01:00.0  On |                  N/A |
    |  0%   38C    P8    10W / 220W |    525MiB /  7982MiB |      0%      Default |
    |                               |                      |                  N/A |
    +-------------------------------+----------------------+----------------------+
                                                                                   
    +-----------------------------------------------------------------------------+
    | Processes:                                                                  |
    |  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
    |        ID   ID                                                   Usage      |
    |=============================================================================|
    |    0   N/A  N/A      1015      G   /usr/lib/xorg/Xorg                 70MiB |
    |    0   N/A  N/A      1542      G   /usr/lib/xorg/Xorg                257MiB |
    |    0   N/A  N/A      1675      G   /usr/bin/gnome-shell               89MiB |
    |    0   N/A  N/A      3560      G   ...AAAAAAAAA= --shared-files       94MiB |
    +-----------------------------------------------------------------------------+
        

我得到以下错误：

print(torch.cuda.current_device())

回溯（最近一次呼叫最后一次）：
文件“”，第1行，在
当前设备中的文件“/home/kumar/anaconda3/envs/pytorch/lib/python3.8/site packages/torch/cuda/__init___.py”，第430行
_lazy_init（）
文件“/home/kumar/anaconda3/envs/pytorch/lib/python3.8/site packages/torch/cuda/__________.py”，第170行，in_lazy__init
火炬._C._cuda_init（）
RuntimeError:CUDA未知错误-这可能是由于环境设置不正确造成的，例如，在程序启动后更改环境变量CUDA_VISIBLE_DEVICES。将可用设备设置为零。

嘿，我也有同样的问题。你设法修好了吗@维诺

print(torch.cuda.current_device())

Traceback (most recent call last):
              File "<stdin>", line 1, in <module>
              File "/home/kumar/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch/cuda/__init__.py", line 430, in current_device
                _lazy_init()
              File "/home/kumar/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch/cuda/__init__.py", line 170, in _lazy_init
                torch._C._cuda_init()
            RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero.