企业官网建设流程全解析

如有侵权或其他问题，欢迎留言联系更正或删除。

1. 运行项目时发现报错：AssertionError: Torch not compiled with CUDA enabled

2. 检测具体错误原因：

运行如下代码：

import torch print(torch.__version__) print(torch.cuda.is_available())

运行结果："+cpu" 表示 torch 版本为 cpu 版本，应为 "+cu" (cuda版本) 且未安装 cuda

3. 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" 改为 "cuda"

8. 解决报错：

原因：计算压力过大，将 dataloader 内的 "n_workers (多线程工作) " 设置为0即可