老师我代码和课上一样,在CPU上运行没问题但GPU上运行后要好久才会报错或反馈。我猜是因为我安装的cuda10.0,而3060显卡要求cuda11.1以上版本?

来源:4-6 利用神经网络解决分类和回归问题(4)

慕娘4345520

2021-10-20

图片1
图片描述
图片2
图片描述图片3
图片描述
图片4
图片描述图片5
图片描述

写回答

1回答

慕娘4345520

提问者

2021-10-21

问题解决了,就是显卡太新,cuda版本不匹配的问题,重新下载配置cuda11.1问题解决

1
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程