torch.cuda.is_available()返回false以及在pytorch官网下载遇到没有符合需求的版本的问题

姜生

于 2022-09-11 13:40:29 发布

阅读量4.8k

点赞数 8

分类专栏：安装记录文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_44658096/article/details/126804455

版权

本文讲述了如何在Python环境中测试PyTorch CUDA可用性，遇到CUDA版本不匹配问题，通过更换源并下载1.10.0版本成功解决。重点在于下载和适配CUDA10.2的PyTorch官方资源。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、在cmd的Python环境下，或者在Anaconda Prompt下输入以下命令进行测试：

import torch
print(torch.__version__)
print(torch.cuda.is_available()) #cuda是否可用，返回为True表示可用
torch.cuda.device_count()#返回GPU的数量
torch