gpt4 book ai didi

pytorch - 断言错误 : Torch not compiled with CUDA enabled (depite several reinstallations)

转载 作者:行者123 更新时间:2023-12-05 01:54:20 26 4
gpt4 key购买 nike

每当我尝试在 pytorch 中将变量移动到 cuda(例如 torch.zeros(1).cuda())时,我都会收到错误消息“AssertionError:Torch 未在启用 CUDA 的情况下编译”。此外,torch.cuda.is_available() 返回 False

我已经阅读了几个解决此错误的答案,但出于某种原因,多次尝试重新安装 cuda 和 putorch 并没有改变任何东西。以下是我使用的一些设置:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

但同样的错误仍然存​​在。可能是什么问题?

一些设置:我使用的是 Ubuntu 20.04,GPU 是 RTX 2080,nvidia-smi 工作正常(NVIDIA-SMI 460.91.03,驱动程序版本:460.91.03,(最大可能)CUDA 版本:11.2)

最佳答案

尝试使用 pip 安装pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html/p>

你可以通过这个线程来获得详细的解释 Pytorch for cuda 11.2

关于pytorch - 断言错误 : Torch not compiled with CUDA enabled (depite several reinstallations),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/70713037/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com