我在 Windows 10 中安装了 GPU 和 CUDA，但 Pytorch 的 torch.cuda.is_avaiable() 返回 false；我该如何纠正？

Question

我在装有 Nvidia GTX 1050 GPU 的 Windows 10 机器上安装了 PyTorch。我已经安装了 CUDA Toolkit 并使用Nvidia 说明对其进行了测试，并且进展顺利，包括执行建议的测试。

但是，torch.cuda.is_available()返回False. 我怎样才能解决这个问题？

Answer 1

我遇到了同样的问题，结果发现我通过运行 https://pytorch.org/get-started/locally/提供的命令安装了仅 CPU 版本。

如果您像我一样安装了 CUDA 10.2，该网站可能会为您提供pip install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html，其中没有明确指定 CPU 或 GPU。

就我而言，它会将我引导至cpu/torch-1.7.1%2Bcpu-cp38-cp38-win_amd64.whlCPU 版本，而不是cu102/torch-1.7.1-cp38-cp38-win_amd64.whl我的 CUDA 10.2。

我的解决方法是，直接转到https://download.pytorch.org/whl/torch_stable.html并下载与您的环境匹配的二进制文件，然后从下载文件夹中简单地安装：

pip install --no-cache-dir --force-reinstall torch===1.7.1 torchvision===0.8.2 -f .\Downloads\

Answer 2

torch.cuda.is_available()造成这种情况的原因是和False的版本不兼容。pytorchcudatoolkit

截至 2022 年 6 月，当前版本的 pytorch 与cudatoolkit=11.3兼容，而当前的cuda 工具包版本 = 11.7。来源

解决方案：

1. 运行conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch安装pytorch。
2. 从https://developer.nvidia.com/cuda-11.3.0-download-archive安装 CUDA 11.3 版本。

你已准备好出发。

原始答案发布在这里： https: //stackoverflow.com/a/72650265/10468354。这仅供快速参考：

Answer 3

我也有同样的问题。并运行这个 => a=torch.cuda.FloatTensor()，给出了断言错误AssertionError: Torch not compiled with CUDA enabled 。...哪种清除了我在没有 cuda 的情况下运行 pytorch。

脚步：

1. 通过调用以下命令确保您已卸载 Pytorch：

   pip uninstall torch

2. 前往https://pytorch.org/get-started/locally/并选择您的系统配置（如图所示）。

3. 从Run this command对话框中复制确切的命令并在终端上运行它。