猿问

“AssertionError:Torch 未在启用 CUDA 的情况下编译”

我发现这是一个流行的问题,但我仍然找不到解决方案。

我正在尝试在这里运行一个简单的 repo ,它使用PyTorch. 虽然我刚刚从 pytorch.org ( 1.2.0) 将我的 Pytorch 升级到了最新的 CUDA 版本,但它仍然会抛出同样的错误。我在 Windows 10 上并将 conda 与 python 3.7 一起使用。

    raise AssertionError("Torch not compiled with CUDA enabled")
    AssertionError: Torch not compiled with CUDA enabled

如何解决问题?


HUWWW
浏览 1492回答 3
3回答

慕丝7291255

你是如何安装pytorch的?听起来您在没有 CUDA 支持的情况下安装了 pytorch。https://pytorch.org/有关于如何安装支持 cuda 的 pytorch 的说明。在这种情况下,我们有以下命令:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch或具有最新 cudatoolkit 版本的命令。

慕容3067478

您不必通过 anaconda 安装它,您可以从他们的网站安装 cuda 。安装结束后打开一个新终端并检查您的 cuda 版本:>>> nvcc --versionnvcc: NVIDIA (R) Cuda compiler driverCopyright (c) 2005-2021 NVIDIA CorporationBuilt on Thu_Nov_18_09:52:33_Pacific_Standard_Time_2021Cuda compilation tools, release 11.5, V11.5.119Build cuda_11.5.r11.5/compiler.30672275_0我的是 V11.5之后,到这里选择你的操作系统和首选的包管理器(pip 或 anaconda),以及你安装的 cuda 版本,然后复制生成的安装命令,我得到:pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html请注意,对我来说,我安装了 python 3.10,但我的项目运行超过 3.9,所以要么使用虚拟环境,要么显式运行你想要的基本解释器的 pip(例如C:\Software\Python\Python39\python.exe -m pip install .....),否则你会遇到Could not find a version that satisfies the requirement torch错误之后,打开 python 控制台并检查 cuda 可用性>>> import torch>>> torch.cuda.is_available()True

MYYA

试试这个:conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
随时随地看视频慕课网APP

相关分类

Python
我要回答