“AssertionError: Torch not compiled with CUDA enabled“的成功解决方法

问题描述
运行代码时遇到:AssertionError: Torch not compiled with CUDA enabled'报错。

错误原因
因为PyTorch没有启用CUDA,而所跑的模型又要求在GPU上运行。

解决方法(前提是设备上有显卡)
方法:首先,检查现用的PyTorch版本是否支持CUDA,以及设备环境变量中是否配置了CUDA路径。若是现用的PyTorch版本不支持CUDA,可以换支持CUDA的PyTorch版本,删除之前已下载的PyTorch版本,重新在命令行终端下载(可选择使用下面的命令重新下载PyTorch)。

# CUDA 11.8
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia

之后,如果下载了支持CUDA的PyTorch版本,但运行代码仍然遇到问题,则要查看电脑上的CUDA路径是否正确配置了,同时检查设备中的显卡是否支持CUDA等其他原因。

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-02-18 20:32:01       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-02-18 20:32:01       106 阅读
  3. 在Django里面运行非项目文件

    2024-02-18 20:32:01       87 阅读
  4. Python语言-面向对象

    2024-02-18 20:32:01       96 阅读

热门阅读

  1. Python 文件IO技巧大全,建议收藏

    2024-02-18 20:32:01       48 阅读
  2. 【模拟】卡拉兹猜想

    2024-02-18 20:32:01       51 阅读
  3. 嵌入式开发工程师的职业发展路径是怎样的?

    2024-02-18 20:32:01       59 阅读
  4. 字符串的解码--leetcode 394

    2024-02-18 20:32:01       58 阅读
  5. 蓝桥杯---奇怪的数列

    2024-02-18 20:32:01       54 阅读
  6. 【Webpack】处理 js 资源

    2024-02-18 20:32:01       54 阅读