关于torch.cuda.is_available()为false

Question

老师好！
torch的版本就用的课程中rag/-requirements.txt中的的版本，nvcc -V 结果release为11.8，nvidia-smi显示 CUDA Version: 12.2，但是 torch.cuda.is_available()为false，应该如何解决？

Answer 1

阿基米口 4天前

1.可以先看下torch是不是gpu版本：

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())

1. torch.version.cuda 为 None 或 cpu 一类 → 说明你装的是 CPU only 版，是用不了 GPU。

2. torch.version.cuda 比如是 11.8 / 11.7 等 → 表示这个 PyTorch 是带 CUDA 的二进制包。

正常情况，按照torch会检测到环境自动按照对应的版本（cpu或者gpu）

如果是cpu版本可以重新安装下：

pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118

• 提问者 qq_慕娘3316381 #1

  老师好，我这边torch.version.cuda 为 None，torch.__version__的结果是2.0.1+cpu，按照老师的方法，结果还是有错，求老师拯救
  (rag) C:\Users\Administrator>pip install pytorch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
  Looking in indexes: https://download.pytorch.org/whl/cu118
  ERROR: Could not find a version that satisfies the requirement pytorch==2.0.1 (from versions: none)
  ERROR: No matching distribution found for pytorch==2.0.1

• 阿基米口 回复 提问者 qq_慕娘3316381 #2

  你可以直接在这里下载对应的whl版本进行安装https://download.pytorch.org/whl/torch_stable.html
  
  cu118/torch-2.0.1%2Bcu118-cp39-cp39-win_amd64.whl
  cu118/torchvision-0.15.2%2Bcu118-cp39-cp39-win_amd64.whl
  
  然后直接pip install xxx.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

• 提问者 qq_慕娘3316381 #3

  回复 qq_慕娘3316381：不好意思老师，已经解决了，应该是pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118

点开查看后面1条评论