torch.cuda.is_available()返回的是FALSE
时间: 2023-11-27 14:06:15 浏览: 104
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
torch.cuda.is_available()返回False可能是因为没有正确安装CUDA或者没有正确安装与CUDA版本相对应的PyTorch。可以按照以下步骤进行排查和解决:
1. 确认CUDA是否正确安装,可以在终端输入nvcc -V查看CUDA版本。
2. 确认PyTorch是否正确安装,可以在Python环境中输入import torch和print(torch.__version__)查看PyTorch版本。
3. 确认PyTorch是否与CUDA版本相对应,可以在PyTorch官网上查看对应版本关系并重新安装相对应的PyTorch。
4. 确认是否正确设置了环境变量,可以在终端输入echo $LD_LIBRARY_PATH查看环境变量是否正确设置。
阅读全文