火炬未与启用CUDA的Windows 10一起编译

时间:2020-07-24 13:51:30

标签: windows pytorch nvidia

我尝试使用 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吗?

0 个答案:

没有答案