关于print(torch.cuda.is_available())输出为False的错误问题

关于print(torch.cuda.is_available())输出为False的错误问题

环境：windows10，python3.9，cuda11.7，torch2.30

问题1. 在命令行使用 nvcc --version 未找到命令，使用nvidia-smi可以显示cuda版本
问题2. 查看cuda和torch版本对应发现torch版本过高

解决1：安装CUDA ToolKit11.7，对应网站：https://developer.nvidia.com/cuda-11-7-0-download-archive, 安装好了之后会发现使用nvcc --version 命令已经可以显示cuda版本

解决2：在自主环境下使用命令对torch进行版本更改处理

相关命令：pip install torch1.13.1+cu117 torchvision0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

再次执行已经显示True

import torch

print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)

1.13.1+cu117
True
11.7