我尝试使用 pytorch 10.2 运行代码,但出现标题中所述的断言错误。首先,我安装了 cuda 版本为 11 。然后我将其卸载并安装了版本 10.2 。另外,我在没有cuda的情况下使用pytorch并使用命令“ conda install pytorch torchvision cpuonly -c pytorch
”
今天安装了Cuda之后,我跑了"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch"
但是,当我运行命令:"torch.cuda.is_available()"
时,输出为False。
运行“ nvcc --version ”后,我看到已安装Cuda 10.2: "Cuda compilation tools, release 10.2, V10.2.89"
。知道如何解决此断言错误吗?
其他答案描述我应该使用cuda安装pytorch,但是我已经这样做了,但仍然遇到相同的错误。我需要手动启用pytorch的cuda吗?