C:\Users\M\.conda\envs\DL\lib\site-packages\torch\cuda\__init__.py:83: UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu. (Triggered internally at C:\actions-runner\_work\pytorch\pytorch\builder\windows\pytorch\c10\cuda\CUDAFunctions.cpp:109.) return torch._C._cuda_getDeviceCount() > 0
时间: 2023-06-30 12:24:41 浏览: 575
【java毕业设计】校内跑腿业务系统源码(springboot+vue+mysql+说明文档).zip
这个警告信息通常表示 PyTorch 初始化 CUDA 失败,原因可能是你的系统没有安装 CUDA 或者没有正确配置 CUDA 相关环境变量。如果你的系统确实没有安装 CUDA,可以尝试在 CPU 上使用 PyTorch;如果你的系统上已经安装了 CUDA,可以检查你的环境变量配置是否正确,比如 `PATH`、`CUDA_HOME` 等。你也可以尝试升级你的显卡驱动或者重装 CUDA 环境。如果你不需要使用 CUDA 计算,可以将 PyTorch 的计算设备设置为 CPU,例如:
```python
import torch
device = torch.device("cpu")
```
这样就可以在 CPU 上运行 PyTorch 程序了。
阅读全文