Cuda 环境配置(windows下)

发布时间:2026/7/20 21:18:08

Cuda 环境配置(windows下) 如有侵权或其他问题欢迎留言联系更正或删除。1. 运行项目时发现报错AssertionError: Torch not compiled with CUDA enabled2. 检测具体错误原因运行如下代码import torch print(torch.__version__) print(torch.cuda.is_available())运行结果cpu 表示 torch 版本为 cpu 版本应为 cu (cuda版本) 且未安装 cuda3. Cuda 的下载、安装、验证参考教程CUDA安装教程超详细-CSDN博客补充提醒Cuda 的安装包需选择 本地 版本选择 线上 版本可能导致安装时出现报错无法连接到 NVIDIA检查您的互联网连接正确操作如下4. cuDNN (Cuda补丁)的下载、安装、验证参考教程CUDA安装教程超详细-CSDN博客5. 重新安装 pytorch需cuda版本参考教程AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】-CSDN博客6. 最终验证全部安装成功。7. 解决报错RuntimeError: CUDA error: invalid device ordinal将配置项内的 cuda:1 改为 cuda8. 解决报错原因计算压力过大将 dataloader 内的 n_workers (多线程工作) 设置为0即可

相关新闻