torch.cuda.is_available() 返回 False 为什么？

Question

有人遇到过这种情况吗？

我尝试更新驱动程序并重新安装cuda

CUDA版本：11.4

图形处理器：GeForce RTX 3060 Laptop(6gb)

操作系统：Windows 10 home

火炬。版本：1.9.0+cpu

Answer 1

您正在使用为 CPU 编译的 PyTorch 版本，您应该安装适当的版本：

• 使用conda：

  conda install pytorch torchvision cudatoolkit=11.1 -c pytorch -c conda-forge 
  

• 使用pip：

  python -m pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html