栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Torch not compiled with CUDA enabled报错的解决办法

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Torch not compiled with CUDA enabled报错的解决办法

Torch not compiled with CUDA enabled报错的解决办法

Windows10+pycharm+Anaconda。需要使用Pytorch,直接在pycharm中安装pytorch,运行后出现如上错误。按照提示不兼容,我查看了CUDA版本,安装指定版本的Pytorch,依旧报错。

错误描述

输入命令print(torch.cuda.is_available())结果为false。
Torch和CUDA不兼容!
处理办法如下:

1查看CUDA版本

查看版本号如上图所示。在 pycharm的Terminal中输入nvcc -V查看CUDA版本。注意英伟达控制面板看到的版本号可能于此不同,以此为准!

2访问PyTorch官网,查找与CUDA版本兼容的Pytorch

  1. 选择你的CUDA版本,将2处的命令复制到pycharm的Terminal中。回车,让其安装。等待安装完成就大功告成。

  2. 如果版本不是上面的选项,例如我的CUDA是9.0.需要进行步骤3.点击进入,选择你的CUDA版本。

  3. 例如我选择通过Conda方式安装 ,系统是Windows,CUDA版本是9.0,选择如上图所示,将4处命令复制到pycharm的Terminal处执行。回车,等待安装完成。

  4. 此时等待安装完成就可以了。除了上面的Conda还可以通过Wheel安装。两种选择一种就可以了!!!

  5. 通过Wheel安装,选择办法如下所示。
    6.

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/834618.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号