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即可