【已解决】pytorch无法检测到cuda

最新推荐文章于 2024-10-05 02:56:00 发布

鹰王eagleking

最新推荐文章于 2024-10-05 02:56:00 发布

阅读量2.1k

点赞数 2

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_44081639/article/details/139703936

版权

一般使用以下变量检查目前pytorch是否检测到了cuda，若为True，表示检测到了，否则就没有。

torch.cuda.is_available()

要用gpu跑模型，就要配置好cuda和cudnn，具体如何配置可以参考：
https://blog.csdn.net/chen565884393/article/details/127905428

如果你已经配置了cuda，但是torch.cuda.is_available()仍然是False，那说明pytorch版本与cuda版本不匹配，这个更新cuda或pytorch的版本：https://blog.csdn.net/u011489887/article/details/135250561

特别的，如果你的cuda是v11.2版本的，则可用以下解决方案：

确保已安装cuda11.2
确保虚拟环境的python版本为python3.8
通过如下命令行，在虚拟环境中用pip安装pytorch1.9.1这个版本

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host pypi.org --trusted-host download.pytorch.org --trusted-host files.pythonhosted.org