pytorch吧 关注:1,269贴子:2,360
  • 4回复贴,共1

求助 配置完环境cuda.is_available()会卡退python

只看楼主收藏回复

nvidia驱动版本556,支持cuda12.5。下载的pytorch2.4,用cuda12.4,一调用cuda.is_available()或者cuda.device_count()就会卡一会儿然后程序中断了,没输出也没报错。去系统日志看会说nvcuda64.dll异常,如图。请问是怎么回事?


计算机是win10,rtx2060


IP属地:广东1楼2024-08-02 18:51回复
    换版本试试


    IP属地:北京2楼2024-08-03 13:26
    回复
      尝试换了低版本的pytorch,还下载了cudatoolkit,但是一样的结果,cuda.is_available()不是输出false,而是调用的时候发生了程序崩溃


      IP属地:广东3楼2024-08-03 15:01
      回复
        那就可能是驱动问题了你一楼发的那个错误的文件我自己的电脑里都没有


        IP属地:北京4楼2024-08-03 19:17
        回复
          源码编译试一下吧,可能是torch或者cuda的bug


          IP属地:内蒙古来自Android客户端5楼2024-08-10 16:52
          回复