level 2
贴吧用户_51XVXX6
楼主
我这cuda，pytorch安装版本基本正确吧？为什么用不了gpu，is_available也是true，还是我环境变量有问题，还是我代码调用gpu的方式错误，还是我把所有库都装在base环境里，造成库冲突之类的![[泪]](/static/emoticons/u6cea.png)
我这是一个简单的mnist手写数字识别代码
2024年12月15日 08点12分
1
我这是一个简单的mnist手写数字识别代码
![[呵呵]](/static/emoticons/u5475u5475.png)
level 11
安装GPU版本的torch
2024年12月17日 12点12分
2
我这2.5.1+cu118不是gpu版本吗
2024年12月17日 13点12分
@贴吧用户_51XVXX6 如果已经安装了GPU版本那就不知道是什么问题了
2024年12月17日 13点12分