pytorch吧 关注:1,412贴子:2,540
  • 3回复贴,共1
求助

求教,正确安装cuda为什么pytorch程序用不了gpu

只看楼主收藏回复

我这cuda,pytorch安装版本基本正确吧?为什么用不了gpu,is_available也是true,还是我环境变量有问题,还是我代码调用gpu的方式错误,还是我把所有库都装在base环境里,造成库冲突之类的我这是一个简单的mnist手写数字识别代码










IP属地:安徽来自Android客户端1楼2024-12-15 16:27回复
    安装GPU版本的torch


    IP属地:北京2楼2024-12-17 20:37
    收起回复