如何解决TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled.

部署运行你感兴趣的模型镜像

原因

原因是因为自己下的对应torch的cuda和计算机本身的cuda版本不一致,导致了Torch not compiled with CUDA enabled,举例来说就是假设计算机本身更新的cuda是12.1的,但是可能由于之前下载pytorch对应的cuda是11.6的,这时候在运行时就会报错。

解决方法

让计算机的cuda和pytorch的cuda版本对应一致。按如下步骤进行:
(1):进入anaconda里自己的的深度学习虚拟环境,conda active xxx,conda list查看pytorch对应的cuda。
在这里插入图片描述
可以看到torch版本对应的是2.3.1 ,cuda是12.1.

(2):看自己NVIDIA能接受cuda的最高型号是多少(后续下载要小于最高型号):第一步点击系统信息,第二部点击组件,以此电脑为例,可以看到允许的最高型号是12.2
在这里插入图片描述
在这里插入图片描述

(3):了解型号之后,去pytorch对应官网下载:复制’Run this Command‘的代码,去anaconda里下载即可。
在这里插入图片描述
(4):再次运行代码。

您可能感兴趣的与本文相关的镜像

PyTorch 2.9

PyTorch 2.9

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

D:\PycharmProjects\nerf1\venv\Scripts\python.exe D:\PycharmProjects\nerf1\run_nerf.py A module that was compiled using NumPy 1.x cannot be run in NumPy 2.0.2 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. Some module may need to rebuild instead e.g. with 'pybind11>=2.12'. If you are a user of the module, the easiest solution will be to downgrade to 'numpy<2' or try to upgrade the affected module. We expect that some modules will need time to support NumPy 2. Traceback (most recent call last): File "D:\PycharmProjects\nerf1\run_nerf.py", line 7, in <module> import torch File "D:\PycharmProjects\nerf1\venv\lib\site-packages\torch\__init__.py", line 870, in <module> from . import _masked File "D:\PycharmProjects\nerf1\venv\lib\site-packages\torch\_masked\__init__.py", line 420, in <module> def sum(input: Tensor, File "D:\PycharmProjects\nerf1\venv\lib\site-packages\torch\_masked\__init__.py", line 223, in _apply_docstring_templates example_input = torch.tensor([[-3, -2, -1], [0, 1, 2]]) D:\PycharmProjects\nerf1\venv\lib\site-packages\torch\_masked\__init__.py:223: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at C:\actions-runner\_work\pytorch\pytorch\builder\windows\pytorch\torch\csrc\utils\tensor_numpy.cpp:68.) example_input = torch.tensor([[-3, -2, -1], [0, 1, 2]]) Traceback (most recent call last): File "D:\PycharmProjects\nerf1\run_nerf.py", line 876, in <module> torch.set_default_tensor_type('torch.cuda.FloatTensor') File "D:\PycharmProjects\nerf1\venv\lib\site-packages\torch\__init__.py", line 323, in set_default_tensor_type _C._set_default_tensor_type(t) TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled. 进程已结束,退出代码为 1
最新发布
06-07
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值