PyCharm运行问题:AssertionError: Torch not compiled with CUDA enabled

这几天在运行代码时出现了以下报错:

File "c:\apps\Miniconda3\lib\site-packages\torch\nn\modules\module.py", line 260, in cuda
return self._apply(lambda t: t.cuda(device))
File "c:\apps\Miniconda3\lib\site-packages\torch\nn\modules\module.py", line 187, in _apply
module._apply(fn)
File "c:\apps\Miniconda3\lib\site-packages\torch\nn\modules\module.py", line 187, in _apply
module._apply(fn)
File "c:\apps\Miniconda3\lib\site-packages\torch\nn\modules\module.py", line 193, in _apply
param.data = fn(param.data)
File "c:\apps\Miniconda3\lib\site-packages\torch\nn\modules\module.py", line 260, in
return self.apply(lambda t: t.cuda(device))
File "c:\apps\Miniconda3\lib\site-packages\torch\cuda_init.py", line 161, in _lazy_init
check_driver()
File "c:\apps\Miniconda3\lib\site-packages\torch\cuda_init.py", line 75, in _check_driver
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

查阅相关资料,说是安装的torch版本是cpu版本,用不了cuda。没办法,在PyCharm的终端中先按照网上的方法使用conda重新安装torch,发现没效果,再使用pip安装torch,还是报错。思来想去,在检查PyCharm的settings时偶然发现PyCharm的torch版本是1.10.0,而我在终端中安装的是1.10.1。于是马上把settings中的torch包去掉,发现PyCharm的工程开始报错没有torch,但是在终端使用

python
import torch
print(torch.__version__)

却依然可以打印出torch版本,说明终端和工程使用的编译器不是同一个python版本!!
再仔细检查,发现终端的python是anaconda环境下的python:

python
Python 3.8.5 (default, Sep  3 2020, 21:29:08) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32

而PyCharm选择的编译器不是anaconda环境下的。
最终,在终端切换python,在编译器的python下使用pip安装torch,解决了问题。

  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
出现AssertionError: Torch not compiled with CUDA enabled错误是因为在编译PyTorch时没有启用CUDA支持。要解决这个问题,你可以按照以下步骤进行操作: 1. 首先,确保你的系统上安装了CUDA驱动程序。你可以从NVIDIA官方网站下载并安装适用于你的显卡的CUDA驱动程序。 2. 接下来,检查你的PyTorch版本是否支持CUDA。你可以通过运行以下代码来检查: ```python import torch print(torch.cuda.is_available()) ``` 如果输出结果为True,表示你的PyTorch支持CUDA。如果输出结果为False,则需要重新安装支持CUDAPyTorch版本。 3. 如果你安装的是CPU版本的PyTorch,而不是支持CUDA的版本,那么你需要重新安装支持CUDAPyTorch。你可以从PyTorch官方网站下载对应的版本,并按照官方文档提供的安装步骤进行操作。 4. 如果你已经安装了支持CUDAPyTorch,但仍然遇到了这个错误,那么可能是因为在安装PyTorch时没有正确设置CUDA相关的环境变量。你可以按照以下步骤进行操作: - 在你的系统环境变量中添加CUDA的路径。具体的路径可能会因为你的系统配置而有所不同,你可以在CUDA安装目录下找到相应的路径。 - 在你的PyCharm项目中设置CUDA相关的环境变量。你可以在PyCharm的设置中找到项目的环境变量设置,并添加CUDA的路径。 通过以上步骤,你应该能够解决AssertionError: Torch not compiled with CUDA enabled错误并成功启用CUDA支持。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值