当前位置:   article > 正文

win11:Torch not compiled with CUDA enabled_win11 assertionerror: torch not compiled with cuda

win11 assertionerror: torch not compiled with cuda enabled

1、我的环境

系统为win11,使用anaconda环境,python版本3.8,已经通过anaconda安装了pytorch 2.0,但是通过命令行nvcc -V以及文件查看发现未安装CUDA

整体思路:先安装的pytorch2.0,然后发现未安装CUDA,安装自己电脑对应的CUDA即11.6版本后,再次运行python程序,发现仍然报错Torch not compiled with CUDA enabled,就去官网https://pytorch.org/get-started/previous-versions/ 查看与自己的cuda对应的版本,卸载当前安装的pytorch,使用官网给出的命令下载pytorch。

一:安装自己电脑对应的CUDA版本

查看CUDA版本

首先桌面上右键->点击选择更多选项->点击NAVID控制面板
打开后:点击帮助,系统信息,组件。这里我的CUDA版本最大支持到11.6.134,我可以安装的CUDA版本不能大于11.6.134,这时我就需要去官网下载11.6.134版本以下的CUDA。
在这里插入图片描述

下载对应的CUDA

输入网址https://developer.nvidia.com/cuda-toolkit-archive,这里我下载11.6.0版本的CUDA。
在这里插入图片描述
接着进入如下界面,进行如下选择。
在这里插入图片描述
然后开始安装。
在这里插入图片描述
然后直接全部下一步。查看是否安装成功,命令行 输入nvcc -V,成功显示如下,则安装成功
在这里插入图片描述

下载对应的cuDNN

可以参考博客:https://blog.csdn.net/jhsignal/article/details/111401628

二:查看与自己的cuda对应的版本

进入官网https://pytorch.org/get-started/previous-versions/
在这里插入图片描述
复制该命令到anaconda的命令里,安装完毕后在anaconda的python环境输入

import torch
print(torch.cuda.isavailable())
  • 1
  • 2

如果成功则显示True,问题就解决了。

三:其他可能的问题

当然我查找问题的时候发现可能存在其他的问题,如:
1,你的pytorch是否安装错了,没有安装GPU版本的
2,anaconda的pytorch和本地的pytorch冲突,参考 https://blog.csdn.net/qq_46941656/article/details/119701547

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/370462
推荐阅读
相关标签
  

闽ICP备14008679号